mindspore.ops.Mish

查看源文件
class mindspore.ops.Mish[源代码]

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

详情请查看 mindspore.ops.mish()

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

    • GPU/CPU:float16、float32、float64。

    • Ascend:float16、float32。

输出:

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

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> 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)
>>> x = Tensor(2.1, mindspore.float32)
>>> output = mish(x)
>>> print(output)
2.050599