Skip to main content

per_step_angle

burn_mamba::mamba3::step_constant

Function per_step_angle 

Source 
fn per_step_angle(rot_ba: Tensor<2>, dt_bh: Tensor<2>) -> Tensor<3>
Expand description

θ̂ = Δ · π·tanh(rot) — the constant per-step RoPE angle increment.