|class||MultiheadAttention< InputDataType, OutputDataType, RegularizerType >|
|Multihead Attention allows the model to jointly attend to information from different representation subspaces at different positions. More...|
Linear algebra utility functions, generally performed on matrices or vectors.
Artificial Neural Network.
Definition of the MultiheadAttention class.
mlpack is free software; you may redistribute it and/or modify it under the terms of the 3-clause BSD license. You should have received a copy of the 3-clause BSD license along with mlpack. If not, see http://www.opensource.org/licenses/BSD-3-Clause for more information.
Definition in file multihead_attention.hpp.