Сравнение тензоров в PyTorch

Если сравнить тензор с числом, с другим тензором или задать сразу несколько условий, на выходе получается маска: в каждой ячейке стоит флаг True или False, показывающий, выполнилось ли условие для этого элемента.

Такой результат снова тензор той же формы, что у исходных данных, а тип значений логический. При выводе на экран ячейки выглядят как tensor([ True, False, True]).

Сравнения задают знаками >, <, >=, <=, == и !=. Несколько условий на один тензор соединяют операторами &, | и ~; каждую часть берут в скобки, чтобы знаки сравнения не смешивались с логикой.

По готовой маске можно подставить в ячейки одни числа там, где условие истинно, и другие, где ложно. Для дробных величин точное равенство через == часто обманчиво из-за округления, поэтому отдельно есть проверка близости двух тензоров.

Со всем этим мы и будем разбираться в рамках данного раздела.