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