cupyx.scipy.special.softmax#

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

Softmax 函数。

Softmax 函数通过计算集合中每个元素的指数，并将其除以所有元素指数之和来转换集合中的每个元素。

参数:

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

  • axis (int 或 tuple of ints, 可选) – 计算值的轴。默认为 None

返回值:

s – 返回与输入形状相同的数组。结果沿指定的轴求和为 1

返回类型:

cupy.ndarray