Mila
Deep Neural Network Library
|
Configuration interface for the MultiHeadAttention module in the Mila DNN framework. More...
Classes | |
class | Mila::Dnn::MultiHeadAttentionConfig |
Configuration class for MultiHeadAttention module. More... | |
Namespaces | |
namespace | Mila |
namespace | Mila::Dnn |
Configuration interface for the MultiHeadAttention module in the Mila DNN framework.
Defines the MultiHeadAttentionConfig class, providing a type-safe fluent interface for configuring MultiHeadAttention modules. Inherits from ComponentConfig CRTP base and adds attention-specific options such as embedding dimension, number of heads, and dropout rates.