Struct MultiHeadSSMConfig 

pub struct MultiHeadSSMConfig {
    pub hidden_dim: usize,
    pub num_heads: usize,
    pub head_dim: usize,
    pub state_dim: usize,
    pub dropout: f32,
    pub causal: bool,
}

Multi-head SSM Attention configuration

Fields

hidden_dim: usize

Model dimension (d_model)

num_heads: usize

Number of attention heads

head_dim: usize

Head dimension (d_model / num_heads)

state_dim: usize

State dimension per head

dropout: f32

Dropout rate

causal: bool

Use causal masking

Implementations

impl MultiHeadSSMConfig

pub fn new( hidden_dim: usize, num_heads: usize, state_dim: usize, ) -> CoreResult<Self>

Create a new configuration

pub fn dropout(self, rate: f32) -> Self

Set dropout rate

pub fn causal(self, causal: bool) -> Self

Set causal masking

Trait Implementations

impl Clone for MultiHeadSSMConfig

fn clone(&self) -> MultiHeadSSMConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MultiHeadSSMConfig

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

Formats the value using the given formatter.