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_valmax_val 进行调整。

Hardtanh函数图:

../../_images/Hardtanh.png
参数:
  • 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.]