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) -> Self

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: Array2<f32>)

Update the state with new values

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

Get the current state

pub fn state_mut(&mut self) -> &mut Array2<f32>

Get mutable reference to state

pub fn step_count(&self) -> usize

Get the step count

pub fn get_row(&self, idx: usize) -> Array1<f32>

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

Formats the value using the given formatter.

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

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::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::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.