Fonction torch.equal de PyTorch
Manuel de référence Pytorch torch
torch.equalC'est une fonction de PyTorch utilisée pour vérifier si deux tenseurs sont égaux. Si les deux tenseurs ont la même forme et les mêmes éléments, elle renvoie True.
Définition de fonction
torch.equal(input, other)
Exemple d'utilisation
Exemple
import torch
# Créer deux tenseurs identiques
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 2, 3])
# Vérifier s'ils sont égaux
result = torch.equal(x, y)
print(result)
# Créer des tenseurs différents
a = torch.tensor([1, 2, 4])
b = torch.tensor([1, 2, 3])
result2 = torch.equal(a, b)
print(result2)
# Créer deux tenseurs identiques
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 2, 3])
# Vérifier s'ils sont égaux
result = torch.equal(x, y)
print(result)
# Créer des tenseurs différents
a = torch.tensor([1, 2, 4])
b = torch.tensor([1, 2, 3])
result2 = torch.equal(a, b)
print(result2)
Le résultat de sortie est:
True False
Autres extensions