mindspore/docs/api/api_python/mint/mindspore.mint.nn.Softmax.rst

9 lines
223 B
ReStructuredText

mindspore.mint.nn.Softmax
=========================
.. py:class:: mindspore.mint.nn.Softmax(dim=None)
对n维输入tensor应用Softmax函数。
更多参考详见 :func:`mindspore.mint.nn.functional.softmax`