Skip to main content

argmax

Function argmax 

Source
fn argmax(logits: &[f32]) -> u32