Skip to main content

argmax

logicaffeine_compile::spatial::decode_engine::model

Function argmax 

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

The greedy token: the index of the largest logit (super::numerics::argmax).