Функция torch.clamp в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.clamp— это функция в PyTorch, используемая для ограничения значений тензора в заданном диапазоне.

Определение функции

torch.clamp(input, min, max, out)

Примеры использования

Пример

import torch

x = torch.tensor([-2, -1, 0, 1, 2, 3])

# Ограничить между -1 и 2
result = torch.clamp(x, min=-1, max=2)

print(result)

Результат вывода:

tensor([-1, -1,  0,  1,  2,  2])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения