Fonction torch.is_nonzero de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.is_nonzeroIl s'agit d'une fonction dans PyTorch utilisée pour vérifier si un tenseur est un élément unique non nul.

Définition de fonction

torch.is_nonzero(input)

Exemple d'utilisation

Exemple

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 参考手册Manuel de référence Pytorch torch

Autres extensions