Функция PyTorch torch.fix
Справочное руководство PyTorch torch
torch.fix— это функция в PyTorch, используемая для поэлементного извлечения целой части (округление в сторону нуля).
Определение функции
torch.fix(input, *, out=None)
Пример использования
Пример
import torch
# Пример извлечения целой части
x = torch.tensor([-1.5, -0.5, 0.5, 1.5, 2.7])
result = torch.fix(x)
print(result) # tensor([-1., -0., 0., 1., 2.])
# Пример извлечения целой части
x = torch.tensor([-1.5, -0.5, 0.5, 1.5, 2.7])
result = torch.fix(x)
print(result) # tensor([-1., -0., 0., 1., 2.])
Другие расширения