Функция clip
Функция clip ограничивает каждый элемент тензора,
приводя его к заданному диапазону. Первым параметром
передаётся тензор, вторым - минимальное значение,
третьим - максимальное. Значения меньше минимума
становятся равными минимуму, а больше максимума -
максимуму.
Синтаксис
torch.clip(input, min, max)
Пример
Ограничим значения тензора диапазоном от 2 до 4:
import torch
t = torch.tensor([1, 3, 5, 7, 9])
res = torch.clip(t, min=2, max=4)
print(res)
Результат выполнения кода:
tensor([2, 3, 4, 4, 4])
Пример
Функция поддерживает работу с числами с плавающей точкой:
import torch
t = torch.tensor([0.2, 0.8, 1.5, 2.3, 3.7])
res = torch.clip(t, min=0.5, max=2.0)
print(res)
Результат выполнения кода:
tensor([0.5000, 0.8000, 1.5000, 2.0000, 2.0000])
Пример
Функция clip может применяться к многомерным тензорам:
import torch
t = torch.tensor([
[1, 5, 9],
[10, 15, 20]
])
res = torch.clip(t, min=3, max=12)
print(res)
Результат выполнения кода:
tensor([
[ 3, 5, 9],
[10, 12, 12]
])
