Hardtanh

class torch.nn.Hardtanh(min_val=-1.0, max_val=1.0, inplace=False, min_value=None, max_value=None) [源代码]

元素级应用HardTanh函数。

HardTanh 的定义是：

$\text{HardTanh}(x) = \begin{cases} \text{max\_val} & \text{ if } x > \text{ max\_val } \\ \text{min\_val} & \text{ if } x < \text{ min\_val } \\ x & \text{ otherwise } \\ \end{cases}$

参数

• min_val (float) – 线性区域范围的最小值，默认为 -1。

• max_val (float) – 线性区域范围的最大值。默认值为 1。

• inplace (bool) – 是否以就地方式执行操作（可选）。默认值： False

关键字参数 min_value 和 max_value 已被弃用，请使用 min_val 和 max_val。

形状:

• 输入: $(*)$，其中$*$表示任意维度的数量。

• 输出: $(*)$，形状与输入相同。

示例:

>>> m = nn.Hardtanh(-2, 2)
>>> input = torch.randn(2)
>>> output = m(input)