argmax
metrale_
model_
weights
1.0.0-beta-preview
In metrale_
model_
weights::
kimi_
k3_
host::
ops
metrale_model_weights
::
kimi_k3_host
::
ops
Function
argmax
Copy item path
Source
pub fn argmax(logits: &[
f32
]) ->
u32
Expand description
Greedy next-token: argmax, lower id on ties.