Функция lt
Функция lt выполняет поэлементное сравнение двух тензоров на условие "меньше". Результатом является тензор с булевыми значениями True или False, где True означает, что соответствующий элемент первого тензора меньше элемента второго тензора. Функция поддерживает сравнение тензоров одинаковой формы, а также сравнение тензора со скаляром. Существует также альтернативный синтаксис через оператор <, который работает аналогично.
Синтаксис
torch.lt(input, other)
Где input и other - тензоры или числа, которые сравниваются поэлементно.
Пример
Давайте сравним два тензора с одинаковой формой на условие "меньше":
import torch
t1 = torch.tensor([1, 3, 5, 7, 9])
t2 = torch.tensor([2, 3, 4, 8, 8])
res = torch.lt(t1, t2)
print(res)
Результат выполнения кода:
tensor([ True, False, False, True, False])
Пример
Используем оператор < для сравнения тензоров - он работает аналогично функции lt:
import torch
t1 = torch.tensor([0, 2, 4, 6, 8])
t2 = torch.tensor([1, 2, 3, 7, 7])
res = t1 < t2
print(res)
Результат выполнения кода:
tensor([ True, False, False, True, False])
Пример
Сравним тензор со скалярным значением:
import torch
t = torch.tensor([1, 5, 3, 8, 2])
res = torch.lt(t, 4)
print(res)
Результат выполнения кода:
tensor([ True, False, True, False, True])
Пример
Используем lt для двумерных тензоров:
import torch
t1 = torch.tensor([
[1, 4, 7],
[2, 5, 8],
])
t2 = torch.tensor([
[2, 4, 6],
[3, 5, 7],
])
res = torch.lt(t1, t2)
print(res)
Результат выполнения кода:
tensor([
[ True, False, False],
[ True, False, False],
])
