cupyx.scipy.special.log_softmax#

cupyx.scipy.special.log_softmax(x, axis=None)[source]#

计算 softmax 函数的对数

参数:

  • x (array-like) – 输入数组

  • axis (int 或 int 的 tuple, 可选) – 计算值的轴。默认为 None，此时将在整个数组 x 上计算 softmax

返回:

s – 形状与 x 相同的数组。结果的指数将沿着指定的轴求和为 1。如果 x 是标量，则返回标量。

返回类型:

cupy.ndarry