jax.scipy.special.log_softmax#

jax.scipy.special.log_softmax(x, /, *, axis=None)[原始碼]#

Log-Softmax 函數。

JAX 實作的 scipy.special.log_softmax()

計算 softmax 函數的對數，該函數將元素重新縮放到 \([-\infty, 0)\) 的範圍。

\[\mathrm{log\_softmax}(x)_i = \log \left( \frac{\exp(x_i)}{\sum_j \exp(x_j)} \right)\]
參數:

  • x (ArrayLike) – 輸入陣列

  • axis (int | tuple[int, ...] | None | None) – 應計算 log_softmax 的軸或軸。

返回:

與 x 形狀相同的陣列

返回類型:

陣列

注意

如果任何輸入值為 +inf，結果將全部為 NaN：這反映了 inf / inf 在浮點數數學的上下文中未明確定義的事實。

參見

softmax()