Skip to main content

neox_order

Function neox_order 

Source
pub fn neox_order(v: &[f64], head_dim: usize, dims: usize) -> Vec<f64>
Expand description

v (whole heads of head_dim) with each head’s rotated dims reordered from interleaved pairs to halves — the basis a View::NeoxPairs-bound program’s queries and keys live in.