cupyx.scipy.special.log_softmax

cupyx.scipy.special.log_softmax(x, axis=None)

计算 softmax 函数的对数

参数:

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

• axis (int or tuple of ints, optional) – 要计算值的轴。默认值为 None，将在整个数组 x 上计算 softmax。

返回:

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

返回类型:

cupy.ndarry