User Guide:
User API:
Developer API:
Full API:
Functions
multi_head_attention_forward
Classes
MultiheadAttention
Allows the model to jointly attend to information from different representation subspaces.