Функция PyTorch torch.is_nonzero

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.is_nonzeroЭто функция в PyTorch, используемая для проверки, является ли тензор одним ненулевым элементом.

Определение функции

torch.is_nonzero(input)

Пример использования

Пример

import torch

print(torch.is_nonzero(torch.tensor([1.0])))
print(torch.is_nonzero(torch.tensor([0.0])))
print(torch.is_nonzero(torch.tensor([1, 2])))

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения