Function scaled_norm_sq

nodedb-codec/src/vector_quant/ternary/codec.rs:43–47

Squared L2 norm of ternary trits scaled by `scale`.

(hot: &[u8], dim: usize, scale: f32)

/// Squared L2 norm of ternary trits scaled by `scale`.
fn scaled_norm_sq(hot: &[u8], dim: usize, scale: f32) -> f32 {
    let trits = unpack_hot(hot, dim);
    let count: i32 = trits.iter().map(|&t| (t != 0) as i32).sum();
    scale * scale * count as f32
}

/// `‖a·sa - b·sb‖² ≈ sa²·‖a‖² + sb²·‖b‖² - 2·sa·sb·dot(a,b)`
fn l2_from_dot(dot: i32, norm_a: f32, norm_b: f32, sa: f32, sb: f32) -> f32 {