split_sinkhorn

spark_model::layers::deepseek_v41_ref::hc

Function split_sinkhorn 

Source 
pub fn split_sinkhorn(
    mixes: &[f32],
    scale: &[f32],
    base: &[f32],
    hc: usize,
    iters: usize,
    eps: f32,
) -> (Vec<f32>, Vec<f32>, Vec<f32>)
Expand description

ds41_ref_shims.hc_split_sinkhorn for one token. mixes: [(2 + hc) * hc]. Returns (pre[hc], post[hc], comb[hc * hc] row-major [j][k]), all f32.