mindspore.mint.nn.functional.mish

mindspore.mint.nn.functional.mish(input)[源代码]

逐元素计算输入Tensor的MISH（A Self Regularized Non-Monotonic Neural Activation Function 自正则化非单调神经激活函数）。

公式如下：

\[\text{mish}(input) = input * \tanh(softplus(\text{input}))\]

更多详细信息请参见 A Self Regularized Non-Monotonic Neural Activation Function 。

Mish激活函数图：

../../_images/Mish.png
参数：

  • input (Tensor) - Mish的输入。支持数据类型：

    • Ascend：float16、float32。

返回：

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

异常：

  • TypeError - input 不是Tensor。

  • TypeError - input 的数据类型不是float16或float32。

支持平台：

Ascend

样例：

>>> import mindspore
>>> from mindspore import Tensor, mint
>>> import numpy as np
>>> x = Tensor(np.array([[-1.1, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32)
>>> output = mint.nn.functional.mish(x)
>>> print(output)
[[-3.0764845e-01 3.9974124e+00 -2.6832507e-03]
 [ 1.9439589e+00 -3.3576239e-02 8.9999990e+00]]