Skip to main content

argmax

Function argmax 

Source
pub fn argmax(values: &[f64]) -> Option<usize>
Expand description

Index of the maximum value (the first on a tie) — the argmax selection used for greedy token decoding (choose the highest-scoring logit). None for an empty slice. Non-finite values compare by the usual > (a NaN never wins).