Пороговое значение

class torch.nn.Threshold(threshold, value, inplace=False) [source]

Применяет пороговое значение к каждому элементу входного тензора.

Пороговое значение определяется как:

$$y = \begin{cases} x, &\text{ если } x > \text{пороговое значение} \\ \text{значение}, &\text{ в противном случае } \end{cases}$$

Параметры

• пороговое значение (float) – Значение порога
• значение (float) – Значение для замены
• inplace (bool) – Необязательно выполнять операцию на месте. По умолчанию: False

Форма:

• Вход: $(*)$, где $*$ обозначает любое количество измерений.
• Выход: $(*)$, такая же форма, как у входных данных.

Примеры:

>>> m = nn.Threshold(0.1, 20)
>>> input = torch.randn(2)
>>> output = m(input)