Функция floor
Функция floor поэлементно округляет числа
в тензоре вниз до ближайшего целого числа.
Функция применяется к каждому элементу тензора
независимо от остальных. Результат возвращается
в виде нового тензора с тем же типом данных или
с типом torch.int64 для целочисленных типов.
Синтаксис
torch.floor(input)
Пример
Давайте создадим тензор с дробными числами и округлим его элементы вниз:
import torch
t = torch.tensor([1.2, 2.7, 3.9, 4.1])
res = torch.floor(t)
print(res)
Результат выполнения кода:
tensor([1., 2., 3., 4.])
Пример
Функция также работает с двумерными тензорами. Каждый элемент округляется вниз независимо:
import torch
t = torch.tensor([
[1.8, 2.2, 3.6],
[4.4, 5.9, 6.1],
])
res = torch.floor(t)
print(res)
Результат выполнения кода:
tensor([
[1., 2., 3.],
[4., 5., 6.],
])
Пример
При работе с отрицательными числами функция округляет их вниз, то есть в сторону отрицательной бесконечности:
import torch
t = torch.tensor([-1.2, -2.7, -3.9, -4.1])
res = torch.floor(t)
print(res)
Результат выполнения кода:
tensor([-2., -3., -4., -5.])
Пример
Если тензор уже содержит целые числа, то функция возвращает их без изменений:
import torch
t = torch.tensor([1, 2, 3, 4])
res = torch.floor(t)
print(res)
Результат выполнения кода:
tensor([1, 2, 3, 4])
