mindspore.ops.HSwish
- class mindspore.ops.HSwish[源代码]
Hard Swish激活函数。
更多参考详见
mindspore.ops.hardswish()
。- 输入:
input_x (Tensor) - 输入Tensor ,其shape为 \((N, *)\) ,其中 \(*\) 表示任意数量的附加维度。
- 输出:
Tensor,数据类型和shape与 input_x 相同。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> hswish = ops.HSwish() >>> input_x = Tensor(np.array([-1, -2, 0, 2, 1]), mindspore.float16) >>> result = hswish(input_x) >>> print(result) [-0.3333 -0.3333 0 1.666 0.6665]