argmax

Function argmax 

Source
pub fn argmax(logits: &[f32]) -> i64
Expand description

Finds the token with the highest logit value (argmax).