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