Skip to main content

split_rotor

burn_mamba::mamba3::rotation

Function split_rotor 

Source 
pub fn split_rotor<const D: usize>(q: Tensor<D>) -> (Tensor<D>, Tensor<D>)
Expand description

Split a stacked RotationState::Rotor accumulator [..., 2·J, 4] into its (left, right) factors, each [..., J, 4].