Функция le
Функция le выполняет поэлементное сравнение "меньше или равно" (≤) между двумя тензорами или между тензором и скаляром. Результатом является тензор с булевыми значениями True или False. Первым параметром передаётся тензор, вторым - тензор или скалярное значение, с которым производится сравнение. Также доступен метод le_, который выполняет операцию на месте.
Синтаксис
torch.le(input, other)
Или как метод тензора:
tensor.le(other)
Пример
Давайте сравним элементы тензора с числом 3 и проверим, какие из них меньше или равны 3:
import torch
t = torch.tensor([1, 3, 5, 2, 4])
res = torch.le(t, 3)
print(res)
Результат выполнения кода:
tensor([True, True, False, True, False])
Пример
Теперь сравним два тензора одинаковой формы поэлементно:
import torch
t1 = torch.tensor([1, 5, 3, 8])
t2 = torch.tensor([2, 4, 3, 7])
res = t1.le(t2)
print(res)
Результат выполнения кода:
tensor([True, False, True, False])
Пример
Использование метода le_ для замены значений тензора на булевы значения на месте:
import torch
t = torch.tensor([10, 20, 30, 40])
t.le_(25)
print(t)
Результат выполнения кода:
tensor([True, True, False, False])
