Struct MambaBlock 

pub struct MambaBlock<B: Backend> { /* private fields */ }

Mamba block for selective state space modeling.

Implementations

impl<B: Backend> MambaBlock<B>

pub fn new(state_dim: usize, input_dim: usize, expand_ratio: usize) -> Self

Create a new Mamba block.

pub fn from_config(config: &MambaConfig) -> Self

Create a Mamba block from configuration.

pub fn state_dim(&self) -> usize

State dimension configured for the block.

pub fn input_dim(&self) -> usize

Input dimension configured for the block.

pub fn expand_ratio(&self) -> usize

Expansion ratio configured for the block.

pub fn forward( &self, input: Tensor<B, 3>, params: &MambaParameters<B>, state: Option<MambaState<B>>, selective: bool, ) -> Result<(Tensor<B, 3>, MambaState<B>), &'static str>

Forward pass for selective state space modeling.

Shapes

• input: [batch, seq_len, input_dim]
• state: [batch, state_dim]

pub fn forward_with_config( &self, input: Tensor<B, 3>, params: &MambaParameters<B>, state: Option<MambaState<B>>, config: &MambaConfig, ) -> Result<(Tensor<B, 3>, MambaState<B>), &'static str>

Forward pass using configuration for selective gating.