Struct SelectiveSSM 

pub struct SelectiveSSM { /* private fields */ }

Selective State Space Model (Mamba-style)

Implements the selective scan mechanism from Mamba for content-aware state transitions.

Implementations

impl SelectiveSSM

pub fn new(config: KizzasiConfig) -> CoreResult<Self>

Create a new SelectiveSSM from configuration

pub fn get_state(&self) -> &HiddenState

Get a reference to the hidden state

pub fn get_state_mut(&mut self) -> &mut HiddenState

Get a mutable reference to the hidden state

pub fn set_state(&mut self, state: HiddenState)

Set the hidden state

pub fn step_count(&self) -> usize

Get the step count from the hidden state

pub fn embedding(&self) -> &ContinuousEmbedding

Get a reference to the embedding layer

pub fn a_matrices(&self) -> &Vec<Array2<f32>>

Get a reference to the A matrices

pub fn b_matrices(&self) -> &Vec<Array2<f32>>

Get a reference to the B matrices

pub fn c_matrices(&self) -> &Vec<Array2<f32>>

Get a reference to the C matrices

pub fn d_vectors(&self) -> &Vec<Array1<f32>>

Get a reference to the D vectors

pub fn output_proj(&self) -> &Array2<f32>

Get a reference to the output projection matrix

Trait Implementations

impl Clone for SelectiveSSM

fn clone(&self) -> SelectiveSSM

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SelectiveSSM

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for SelectiveSSM

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for SelectiveSSM

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl SignalPredictor for SelectiveSSM

fn step(&mut self, input: &Array1<f32>) -> CoreResult<Array1<f32>>

Update hidden state and predict the next signal vector

fn reset(&mut self)

Reset the hidden state to initial values

fn context_window(&self) -> usize

Get the current context window size

impl StateSpaceModel for SelectiveSSM

fn recurrence_step( &self, input: &Array1<f32>, state: &mut HiddenState, ) -> CoreResult<Array1<f32>>

Perform a single recurrence step

fn config(&self) -> &KizzasiConfig

Get model configuration