mindspore.ops.mish

mindspore.ops.mish(x)[源代码]

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

公式如下：

\[\text{output} = x * \tanh(\log(1 + \exp(\text{x})))\]

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

Mish函数图：

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

  • x (Tensor) - 输入Tensor。 支持数据类型：

    • GPU/CPU：float16、float32、float64。

    • Ascend：float16、float32。

返回：

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

异常：

  • TypeError - x 的数据类型不是float16、float32或float64。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input_x = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32)
>>> output = ops.mish(input_x)
>>> print(output)
[[-3.0340147e-01  3.9974129e+00 -2.68311895e-03]
 [ 1.9439590e+00  -3.3576239e-02 8.99999990e+00]]
>>> input_x = Tensor(2.1, mindspore.float32)
>>> output = ops.mish(input_x)
>>> print(output)
2.050599