Struct SurrogateLif 

pub struct SurrogateLif {
    pub lif: FixedPointLif,
    pub surrogate: SurrogateType,
    membrane_trace: Vec<(f32, f32)>,
}

LIF neuron with surrogate gradient support.

Fields

lif: FixedPointLif

surrogate: SurrogateType

membrane_trace: Vec<(f32, f32)>

Implementations

impl SurrogateLif

pub fn new( data_width: u32, fraction: u32, v_rest: i16, v_reset: i16, v_threshold: i16, refractory_period: i32, surrogate: SurrogateType, ) -> Self

Construct a surrogate-enabled fixed-point LIF neuron.

pub fn forward( &mut self, leak_k: i16, gain_k: i16, i_t: i16, noise_in: i16, ) -> (i32, i16)

Forward LIF step while caching trace for backward pass.

pub fn backward(&mut self, grad_output: f32) -> Result<f32, String>

Backward pass through last cached membrane value.

pub fn clear_trace(&mut self)

Clear stored membrane trace.

pub fn reset(&mut self)

Reset neuron state and clear trace.

pub fn trace_len(&self) -> usize

Number of cached forward steps.