mindspore.nn.HSigmoid
- class mindspore.nn.HSigmoid[源代码]
Hard Sigmoid激活函数。按元素计算输出。
Hard Sigmoid定义为:
\[\text{hsigmoid}(x_{i}) = max(0, min(1, \frac{x_{i} + 3}{6})),\]- 输入:
input_x (Tensor) - Hard Sigmoid的输入,任意维度的Tensor。
- 输出:
Tensor,数据类型和shape与 input_x 的相同。
- 异常:
TypeError - input_x 不是tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> x = Tensor(np.array([-1, -2, 0, 2, 1]), mindspore.float16) >>> hsigmoid = nn.HSigmoid() >>> result = hsigmoid(x) >>> print(result) [0.3333 0.1666 0.5 0.8335 0.6665]