Function logits 

pub fn logits(model: &K3CpuModel, h: &[f32]) -> Vec<f32>

LM head over a hidden vector. W is [vocab, hidden].