jax.scipy.special.softmax

jax.scipy.special.softmax(x, /, *, axis=None)

Softmax 函数。

JAX 实现的 scipy.special.softmax()。

计算将元素重新缩放到范围 \([0, 1]\) 的函数，使得沿 轴 的元素之和为 \(1\)。

\[\mathrm{softmax}(x) = \frac{\exp(x_i)}{\sum_j \exp(x_j)}\]

参数:

• x (ArrayLike) – 输入数组

• axis (int | tuple[int, ...] | None) – softmax 应该计算的轴或轴。在这些维度上 softmax 输出的总和应为 \(1\)。

返回:

与 x 形状相同的数组。

返回类型:

Array

备注

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

参见

log_softmax()