Метод bool
Метод bool преобразует тензор к логическому типу данных (torch.bool).
При преобразовании все ненулевые элементы становятся True, а нулевые - False.
Метод возвращает новый тензор с логическими значениями, не изменяя исходный тензор.
Синтаксис
tensor.bool()
Пример
Давайте преобразуем целочисленный тензор в логический тип:
import torch
t = torch.tensor([0, 1, 2, 0, 3])
res = t.bool()
print(res)
Результат выполнения кода:
tensor([False, True, True, False, True])
Пример
Преобразуем тензор с отрицательными числами:
import torch
t = torch.tensor([-1, 0, 5, -3, 0])
res = t.bool()
print(res)
Результат выполнения кода:
tensor([ True, False, True, True, False])
Пример
Преобразуем двумерный тензор в логический тип:
import torch
t = torch.tensor([
[0, 1, 0],
[2, 0, 3],
])
res = t.bool()
print(res)
Результат выполнения кода:
tensor([
[False, True, False],
[ True, False, True],
])
Пример
Метод bool часто используется для создания маски из условий:
import torch
t = torch.tensor([1, 2, 3, 4, 5])
mask = (t > 2).bool()
print(mask)
Результат выполнения кода:
tensor([False, False, True, True, True])
