pub fn token_nll(logits: &[f32], target: usize) -> f64
Negative log-likelihood (nats) of target under softmax(logits) = logsumexp - logits[target].
target
softmax(logits)
logsumexp - logits[target]