std1.0.0-beta-7 SafeC docs

std :: ml::ssm

Struct std::MambaState

#include <std/ml/ssm.h> source

struct MambaState {
    &FloatTensor conv;
    &FloatTensor h;
}

Recurrent decoding, one token at a time with a fixed-size state. Feeding tokens 0..T-1 through Mamba::step gives the same rows as Mamba::forward on the whole sequence.

Fields

[dInner, dConv - 1], the last inputs to the convolution, oldest first

[dInner, dState], the SSM state

Methods

§void reset()source · definition
§void drop()source · definition
§void free() consumesource · definition

= drop()

§static struct MambaState new(const &Mamba layer)source · definition

zero state