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) – 计算 log_softmax 应该沿着哪个轴或哪些轴。

返回:

x 的形状与 x 相同的数组

返回类型:

Array

注意

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

另请参阅

softmax()