pub fn argmax(logits: &[f32]) -> SimdResult<usize>Expand description
Computes argmax over a slice of logits using the best available implementation. Returns the index of the maximum value.
pub fn argmax(logits: &[f32]) -> SimdResult<usize>Computes argmax over a slice of logits using the best available implementation. Returns the index of the maximum value.