How be to use the encoder and decoder of transformer in another model for tf2.0?

[yang182]

Description

If I have another model, but in sub-layer of my model, I want to add a transformer-encoder layer and add a tranformer-decoder or multi transformer-decoder in another layer. How can I do? If my model is tensorflow2.0, could it work? ...

Environment information

tf2.0

OS: <your answer here>

$ pip freeze | grep tensor
# your output here

$ python -V
# your output here

For bugs: reproduction and error logs

# Steps to reproduce:
...

# Error logs:
...