mindspore.ops.hardsigmoid
- mindspore.ops.hardsigmoid(input)[源代码]
Hard Sigmoid激活函数。按元素计算输出。
Hard Sigmoid定义为:
\[\begin{split}\text{Hardswish}(input) = \begin{cases} 0, & \text{ if } input \leq -3, \\ 1, & \text{ if } input \geq +3, \\ input/6 + 1/2, & \text{ otherwise } \end{cases}\end{split}\]HSigmoid函数图:
- 参数:
input (Tensor) - 输入Tensor。
- 返回:
Tensor,shape和数据类型与输入 input 相同。
- 异常:
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.hardsigmoid(input) >>> print(output) [0.3333 0.1666 0.5 0.8335 0.6665]