Сравнение двух тензоров в PyTorch

Два тензора одной формы сравнивают попарно: первая ячейка с первой, вторая со второй и так далее. Знак сравнения даёт логический тензор той же формы.

Создадим два ряда и проверим, где первый строго больше второго:

import torch

left = torch.tensor([1, 5, 3])
right = torch.tensor([2, 4, 3])
print(left > right) # выведет tensor([False,  True, False])

Равенство по ячейкам проверяют тем же способом:

import torch

left = torch.tensor([1, 5, 3])
right = torch.tensor([1, 4, 3])
print(left == right) # выведет tensor([ True, False,  True])

Для таблиц правило то же, если число рядов и столбцов совпадает:

import torch

a = torch.tensor([[1, 6], [3, 2]])
b = torch.tensor([[1, 4], [5, 2]])
print(a >= b) # выведет tensor([[ True,  True], [False,  True]])

Создайте два ряда из трёх целых чисел и выведите маску, где первый ряд строго меньше второго по каждой позиции.

Задайте два ряда одинаковой длины и выведите маску неравенства по соответствующим ячейкам.

Создайте две таблицы 2 на 2 и выведите маску, где в каждой ячейке значения в таблицах совпадают.