mindspore.ops.LogSoftmax

class mindspore.ops.LogSoftmax(axis=- 1)

LogSoftmax激活函数。

在指定轴上对输入Tensor应用LogSoftmax函数。假设在指定轴上， \(x\) 对应每个元素 \(x_i\) ，则LogSoftmax函数如下所示：

\[\text{output}(x_i) = \log \left(\frac{\exp(x_i)} {\sum_{j = 0}^{N-1}\exp(x_j)}\right),\]

其中， \(N\) 为Tensor长度。

参数：

• axis (int) - 指定进行运算的轴。默认值：–1。

输入：

• logits (Tensor) - LogSoftmax的输入，任意维度的Tensor，其数据类型为float16或float32。

输出：

Tensor，数据类型和shape与 logits 相同。

异常：

• TypeError - axis 不是int。

• TypeError - logits 的数据类型既不是float16也不是float32。

• ValueError - axis 不在[-len(logits.shape), len(logits.shape))范围中。

支持平台：

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]