Skip to main content

softmax

Function softmax 

Source
pub fn softmax(logits: &[f64], temperature: f64) -> Result<Vec<f64>>
Expand description

ロジット列から温度付きソフトマックス確率分布を計算し、新しいベクタとして返却する。

§概要

  • 内部でゼロアロケーション版関数 softmax_into を呼び出す。
  • 返却される確率ベクタの総和はほぼ 1.0 となり、各要素は [0.0, 1.0] に収まる。

§引数

  • logits: 未正規化の生ロジットスライス(f64)。
  • temperature: 較正温度パラメータ $\tau$。正の有限実数。

§戻り値

  • 成功時は計算された確率分布ベクタ(Vec<f64>)、異常時は CoreError::MathError。