Сравнение с числом в PyTorch

Число участвует в сравнении с каждой ячейкой ряда или таблицы, как будто оно записано в той же позиции. На выходе получается логический тензор той же формы: True там, где условие выполнилось, и False в остальных ячейках.

Создадим ряд оценок и проверим, какие элементы больше 3:

import torch

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

Те же данные сравним с нулём на равенство и на «не больше трёх»:

import torch

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

Знак != отмечает ячейки, где значение не совпадает с заданным числом:

import torch

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

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

Задайте ряд из четырёх температур и выведите маску, где значение не меньше 0.

Создайте ряд из чисел 1, 2, 2 и 3 и выведите маску совпадения с 2.