mindspore.ops.Mish
- class mindspore.ops.Mish[源代码]
逐元素计算输入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 。
- 输入:
x (Tensor) - shape: \((N, *)\) ,其中 \(*\) 表示任意数量的附加维度,数据类型支持float16或float32。
- 输出:
Tensor,与 x 的shape和数据类型相同。
- 异常:
TypeError - x 的数据类型非float16或float32。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> x = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32) >>> mish = ops.Mish() >>> output = mish(x) >>> print(output.shape) (2, 3)