Fonction torch.ne de PyTorch
Manuel de référence Pytorch torch
torch.neest une fonction dans PyTorch utilisée pour la comparaison d'inégalité élément par élément.
Définition de fonction
torch.ne(input, other, out)
Exemple d'utilisation
Exemple
import torch
a = torch.tensor([1, 2, 3, 4])
b = torch.tensor([1, 2, 0, 4])
result = torch.ne(a, b)
print(result)
a = torch.tensor([1, 2, 3, 4])
b = torch.tensor([1, 2, 0, 4])
result = torch.ne(a, b)
print(result)
Le résultat de sortie est:
tensor([False, False, True, False])
Autres extensions