Struct ShortConv 

pub struct ShortConv { /* private fields */ }

Short convolution for SSM (commonly kernel_size=4 in Mamba)

Optimized implementation for small kernel sizes using loop unrolling.

Implementations

impl ShortConv

pub fn new(channels: usize) -> Self

Create a new short convolution (defaults to kernel_size=4)

pub fn with_kernel_size(channels: usize, kernel_size: usize) -> Self

Create with custom kernel size

pub fn forward(&mut self, input: &Array1<f32>) -> Array1<f32>

Forward pass

pub fn reset(&mut self)

Reset state

pub fn set_weights(&mut self, weights: Vec<Vec<f32>>)

Set weights

pub fn channels(&self) -> usize

Get channels

Trait Implementations

impl Clone for ShortConv

fn clone(&self) -> ShortConv

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ShortConv

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

Formats the value using the given formatter.