Порог

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)