mindspore.mint.nn.Softmax

class mindspore.mint.nn.Softmax(dim=None)

将Softmax函数应用于n维输入Tensor。

更多参考详见 mindspore.mint.nn.functional.softmax()。

支持平台：

Ascend

样例：

>>> import mindspore
>>> from mindspore import Tensor, mint
>>> import numpy as np
>>> input = Tensor(np.array([-1, -2, 0, 2, 1]), mindspore.float16)
>>> softmax = mint.nn.Softmax()
>>> output = softmax(input)
>>> print(output)
[0.03168 0.01166 0.0861  0.636   0.2341 ]