mindspore.ops.hardtanh
- mindspore.ops.hardtanh(input, min_val=- 1.0, max_val=1.0)[源代码]
逐元素元素计算hardtanh激活函数。hardtanh函数定义为:
\[\begin{split}\text{hardtanh}(input) = \begin{cases} max\_val, & \text{ if } input > max\_val \\ min\_val, & \text{ if } input < min\_val \\ input, & \text{ otherwise. } \end{cases}\end{split}\]线性区域范围 \([min\_val, max\_val]\) 可以使用 min_val 和 max_val 进行调整。
Hardtanh函数图:
- 参数:
input (Tensor) - 输入Tensor。
min_val (Union[int, float],可选) - 线性区域范围的最小值。默认值:
-1.0
。max_val (Union[int, float],可选) - 线性区域范围的最大值。默认值:
1.0
。
- 返回:
Tensor,数据类型和shape与 input 的相同。
- 异常:
TypeError - input 不是Tensor。
TypeError - min_val 的数据类型既不是int也不是float。
TypeError - max_val 的数据类型既不是int也不是float。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> from mindspore import Tensor, ops >>> x = Tensor([-1, -2, 0, 2, 1], mindspore.float16) >>> output = ops.hardtanh(x, min_val=-1.0, max_val=1.0) >>> print(output) [-1. -1. 0. 1. 1.]