diff --git a/src/mlpack/methods/ann/layer/multihead_attention.hpp b/src/mlpack/methods/ann/layer/multihead_attention.hpp index 3937ad5530..15421566ce 100644 --- a/src/mlpack/methods/ann/layer/multihead_attention.hpp +++ b/src/mlpack/methods/ann/layer/multihead_attention.hpp @@ -74,7 +74,8 @@ class MultiheadAttentionType : public Layer * @param numHeads Number of parallel attention heads. * @param attnMask Two dimensional Attention Mask. * @param keyPaddingMask Key Padding Mask. - * @param selfAttention Use self-attention; source key, query, and value all come from the same inputs + * @param selfAttention Use self-attention; source key, query, and value all + * come from the same inputs */ MultiheadAttentionType(const size_t tgtSeqLen, const size_t srcSeqLen,