Функция torch.round в PyTorch
Справочное руководство Pytorch torch
torch.roundЭто функция в PyTorch для поэлементного округления.
Определение функции
torch.round(input, decimals, out)
Пример использования
Пример
import torch
x = torch.tensor([0.1, 0.5, 0.9, 1.2, 1.5])
print(torch.round(x))
# Укажите количество десятичных знаков
y = torch.tensor([1.234, 5.678])
print(torch.round(y, decimals=2))
x = torch.tensor([0.1, 0.5, 0.9, 1.2, 1.5])
print(torch.round(x))
# Укажите количество десятичных знаков
y = torch.tensor([1.234, 5.678])
print(torch.round(y, decimals=2))
Результат вывода:
tensor([0., 0., 1., 1., 2.]) tensor([1.2300, 5.6800])
Другие расширения