Struct DecoderConfig 

pub struct DecoderConfig {
    pub self_attention: AttentionConfig,
    pub cross_attention: AttentionConfig,
    pub feed_forward: FeedForwardConfig,
    pub layer_norm: LayerNormConfig,
    pub pre_norm: bool,
}

Configuration for transformer decoder layer

Fields

self_attention: AttentionConfig

Self-attention configuration (with causal masking)

cross_attention: AttentionConfig

Cross-attention configuration

feed_forward: FeedForwardConfig

Feed-forward configuration

layer_norm: LayerNormConfig

Layer normalization configuration

pre_norm: bool

Whether to use pre-layer normalization

Implementations

impl DecoderConfig

pub fn new(d_model: usize, n_heads: usize, d_ff: usize) -> Result<Self>

Create a new decoder configuration

pub fn with_pre_norm(self, pre_norm: bool) -> Self

Set pre-normalization vs post-normalization

pub fn with_dropout(self, dropout: f64) -> Self

Set dropout

pub fn validate(&self) -> Result<()>

Validate configuration

Trait Implementations

impl Clone for DecoderConfig

fn clone(&self) -> DecoderConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DecoderConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.