multihead_attention.hpp File Reference
Include dependency graph for multihead_attention.hpp:

Go to the source code of this file.

Classes

class  MultiheadAttentionType< InputType, OutputType, RegularizerType >
 Multihead Attention allows the model to jointly attend to information from different representation subspaces at different positions. More...

 

Namespaces

 mlpack
 
Linear algebra utility functions, generally performed on matrices or vectors.
 
 mlpack::ann
 
Artificial Neural Network.
 

Typedefs

typedef MultiheadAttentionType< arma::mat, arma::mat, NoRegularizer > MultiheadAttention