Fonction torch.eq de PyTorch
Manuel de référence Pytorch torch
torch.eqest une fonction de PyTorch utilisée pour la comparaison d'égalité élément par élément.
Définition de fonction
torch.eq(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.eq(a, b)
print(result)
a = torch.tensor([1, 2, 3, 4])
b = torch.tensor([1, 2, 0, 4])
result = torch.eq(a, b)
print(result)
Le résultat de sortie est:
tensor([True, True, False, True])
Autres extensions