Skip to main content

softmax

zsfm_core

Function softmax 

Source 
pub fn softmax(logits: &[f32]) -> Vec<f32>
Expand description

Numerically stable softmax over host f32 logits.