torch.nn.functional.threshold_

torch.nn.functional.threshold_(input, threshold, value) → Tensor

Функция замены значений в тензоре, если они меньше или равны порогу. Вместо threshold() используется модификация самого тензора.