Expand description
Re-exports§
pub use self::attention::scaled_dot_product_attention;pub use self::attention::AttentionHead;pub use self::params::*;
Modules§
Structs§
Constants§
- DK
- The default dimension of the key and query vectors
- D_MODEL
- The default dimension of the model; i.e. the number of inputs
- D_
NETWORK - The default size of the network; i.e. the number of neurons in the network
- HEADS
- The default number of attention heads
- N
- The default number of layers used for the encoder / decoder.