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)

返回：

一个数组。

返回类型：

数组

注意

如果任何输入值是 +inf，结果将全部为 NaN：这反映了在浮点数学的上下文中，inf / inf 没有明确定义的这一事实。

另请参阅

softmax()