mindspore.ops.mish
- mindspore.ops.mish(x)[源代码]
逐元素计算输入Tensor的MISH(A 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激活函数图:
- 参数:
x (Tensor) - Mish的输入。 支持数据类型:
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