Функция PyTorch torch.floor

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

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

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

torch.floor(input, out=None)

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

Пример

import torch

x = torch.tensor([1.1, 2.5, 3.7, -1.2])

print(torch.floor(x))

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

tensor([1., 2., 3., -2.])

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

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