Struct HiddenState 

pub struct HiddenState { /* private fields */ }

Represents the hidden state of the SSM

Implementations

impl HiddenState

pub fn new(hidden_dim: usize, state_dim: usize) -> HiddenState

Create a new hidden state with given dimensions

pub fn reset(&mut self)

Reset the state to zeros

pub fn set_conv_history(&mut self, history: Vec<Vec<f32>>)

Set the convolution history

pub fn conv_history(&self) -> Option<&Vec<Vec<f32>>>

Get the convolution history

pub fn take_conv_history(&mut self) -> Option<Vec<Vec<f32>>>

Take the convolution history (moves ownership)

pub fn update(&mut self, new_state: ArrayBase<OwnedRepr<f32>, Dim<[usize; 2]>>)

Update the state with new values

pub fn state(&self) -> &ArrayBase<OwnedRepr<f32>, Dim<[usize; 2]>>

Get the current state

pub fn state_mut(&mut self) -> &mut ArrayBase<OwnedRepr<f32>, Dim<[usize; 2]>>

Get mutable reference to state

pub fn step_count(&self) -> usize

Get the step count

pub fn get_row(&self, idx: usize) -> ArrayBase<OwnedRepr<f32>, Dim<[usize; 1]>>

Get a row of the state as 1D array

Trait Implementations

impl Clone for HiddenState

fn clone(&self) -> HiddenState

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for HiddenState

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for HiddenState

fn deserialize<__D>( __deserializer: __D, ) -> Result<HiddenState, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for HiddenState

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

where __S: Serializer,

Serialize this value into the given Serde serializer.