Python PyTorch Hardtanh用法及代码示例

本文简要介绍python语言中 torch.nn.Hardtanh 的用法。

用法:

class torch.nn.Hardtanh(min_val=- 1.0, max_val=1.0, inplace=False, min_value=None, max_value=None)

参数：

• min_val-线性区域范围的最小值。默认值：-1

• max_val-线性区域范围的最大值。默认值：1

• inplace-可以选择就地执行操作。默认值：False

按元素应用HardTanh 函数

HardTanh 定义为：

\text{HardTanh}(x) = \begin{cases} 1 & \text{ if } x > 1 \\ -1 & \text{ if } x < -1 \\ x & \text{ otherwise } \\ \end{cases}

可以使用 min_val 和 max_val 调整线性区域 [-1, 1] 的范围。

关键字参数 min_value 和 max_value 已被弃用，取而代之的是 min_val 和 max_val 。

形状：

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

• 输出：(*)，与输入的形状相同。

例子：

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