快捷方式

Hardtanh

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

逐元素应用 HardTanh 函数。

HardTanh 定义为

HardTanh(x)={max_val if x> max_val min_val if x< min_val x otherwise \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_valuemax_value 已被弃用,建议使用 min_valmax_val

形状
  • 输入: ()(*), 其中 * 表示任意数量的维度。

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

../_images/Hardtanh.png

示例

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

文档

访问 PyTorch 的全面开发人员文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得您的问题解答

查看资源