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