mindspore.nn.HSigmoid
- class mindspore.nn.HSigmoid[源代码]
逐元素计算Hard Sigmoid激活函数。
Hard Sigmoid定义为:
\[\text{hsigmoid}(x_{i}) = \max(0, \min(1, \frac{x_{i} + 3}{6})),\]HSigmoid函数图:
- 输入:
input_x (Tensor) - Hard Sigmoid的输入,任意维度的Tensor。
- 输出:
Tensor,数据类型和shape与 input_x 的相同。
- 异常:
TypeError - input_x 不是Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> from mindspore import Tensor, nn >>> import numpy as np >>> 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]