pub fn score_matrix(
q: &[f32],
k: &[f32],
m: usize,
n: usize,
d: usize,
scores: &mut [f32],
)Expand description
Compute scaled dot-product score matrix: scores[i,j] = Q[i] . K[j] / sqrt(d).
Q is m x d, K is n x d, scores is m x n (row-major).