Skip to main content

moments_phys_fwd

Function moments_phys_fwd 

Source
pub(crate) fn moments_phys_fwd<B: Backend>(
    xhat_bnlchp: F<B, 6>,
    bhat_bnlchr: F<B, 6>,
    da_bnlh: F<B, 4>,
    rot: &RotPrim<B>,
    initial_state_bhpr: F<B, 4>,
    valid_len: usize,
) -> (F<B, 4>, F<B, 3>)
Expand description

The serial chunkwise forward on primitives — the exact math of Mamba3MomentsInput::state_moments_phys.

Returns (m2_bhrr, m1_bhr).