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

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> 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]