mindspore.ops.hardswish
- mindspore.ops.hardswish(input)[源代码]
Hard Swish激活函数。输入是一个Tensor,具有任何有效的shape。
Hard Swish定义如下:
\[\begin{split}\text{Hardswish}(input) = \begin{cases} 0, & \text{ if } input \leq -3, \\ input, & \text{ if } input \geq +3, \\ input*(input + 3)/6, & \text{ otherwise } \end{cases}\end{split}\]HSwish函数图:
- 参数:
input (Tensor) - Hard Swish的输入。
- 返回:
Tensor,shape和数据类型与输入相同。
- 异常:
TypeError - input 不是一个Tensor。
TypeError - input 不是int或者float类型。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> input = Tensor(np.array([-1, -2, 0, 2, 1]), mindspore.float16) >>> output = ops.hardswish(input) >>> print(output) [-0.3333 -0.3333 0 1.667 0.6665]