Hardtanh

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

Применяет функцию 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)