state-spaces / mamba

Mamba SSM architecture
Apache License 2.0
12.7k stars 1.06k forks source link

I'm wondering if this final_state refers to the state depicted by the red circle in the figure #557

Open jialiangZ opened 2 weeks ago

jialiangZ commented 2 weeks ago

out, x, *rest = selective_scan_cuda.fwd(u, delta, A, B, C, D, z, delta_bias, delta_softplus) image I'm wondering if this final_state refers to the state depicted by the red circle in the figure