jax.nn.log_softmax

jax.nn.log_softmax(x, axis=-1, where=None, initial=_UNSPECIFIED)

对数 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 (ArrayLike | None | Unspecified)

返回:

数组。

返回类型:

数组

注意

如果任何输入值为 +inf，则结果将全部为 NaN：这反映了 inf / inf 在浮点数学中没有定义。

另请参阅

softmax()