jax.nn.log_softmax#

jax.nn.log_softmax(x, axis=-1, where=None, initial=_UNSPECIFIED)[原始碼]#

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) – 應該計算 log_softmax 的軸或軸。可以是整數或整數元組。

  • where (ArrayLike | None) – 要包含在 log_softmax 中的元素。

  • initial (Unspecified)

返回:

一個陣列。

返回類型:

Array

注意

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

參見

softmax()