Enum RotPrim 

pub(crate) enum RotPrim<B: Backend> {
    Complex {
        cum_bsha: F<B, 4>,
        rope_dim: usize,
        rotate_pairwise: bool,
    },
    Quaternion {
        cum_bshj4: F<B, 5>,
    },
}

The per-token cumulative rotation on B’s primitives — the rank-tagged counterpart of RotationSeq.

Variants

Complex

Cumulative angles [batch, sequence, nheads, num_rope_angles].

Fields

cum_bsha: F<B, 4>

rope_dim: usize

rotate_pairwise: bool

Quaternion

Cumulative unit quaternions [batch, sequence, nheads, blocks, 4].

Fields

cum_bshj4: F<B, 5>

Implementations

impl<B: Backend> RotPrim<B>

pub fn wrap( rot: FloatTensor<B>, quaternion: bool, rope_dim: usize, rotate_pairwise: bool, ) -> Self

Wrap the rank-erased seam arguments.