Функция torch.clip в PyTorch
Справочное руководство PyTorch torch
torch.clipявляется функцией в PyTorch, используемой для ограничения значений тензора до указанного диапазона.
Определение функции
torch.clip(input, min, max, out=None)
Примеры использования
Пример
import torch
# Создать тензор
x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# Ограничить значения диапазоном [2, 4]
y = torch.clip(x, 2, 4)
print(y)
# Создать тензор
x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# Ограничить значения диапазоном [2, 4]
y = torch.clip(x, 2, 4)
print(y)
Результат вывода:
tensor([2., 2., 3., 4., 4.])
Другие расширения