mindspore.ops.LogSoftmax
- class mindspore.ops.LogSoftmax(axis=- 1)[源代码]
LogSoftmax激活函数。
更多参考详见
mindspore.ops.log_softmax()
。- 参数:
axis (int,可选) - 指定进行运算的轴。默认值:-1。
- 输入:
logits (Tensor) - shape: \((N, *)\) ,其中 \(*\) 表示任意数量的附加维度,其数据类型为float16或float32。
- 输出:
Tensor,数据类型和shape与 logits 相同。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> logits = Tensor(np.array([1, 2, 3, 4, 5]), mindspore.float32) >>> log_softmax = ops.LogSoftmax() >>> output = log_softmax(logits) >>> print(output) [-4.4519143 -3.4519143 -2.4519143 -1.4519144 -0.4519144]