Fonction torch.allclose de PyTorch
Manuel de référence PyTorch torch
torch.allcloseUne fonction de PyTorch utilisée pour vérifier si tous les éléments de deux tenseurs sont proches. Elle sert à déterminer si deux tenseurs sont égaux à une tolérance donnée près.
Définition de fonction
torch.allclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False)
Exemple d'utilisation
Exemple
import torch
# Créer deux tenseurs similaires
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([1.0, 2.0, 3.0 + 1e-9])
# Vérifier si tous les éléments sont proches
result = torch.allclose(x, y)
print(result)
# Créer deux tenseurs similaires
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([1.0, 2.0, 3.0 + 1e-9])
# Vérifier si tous les éléments sont proches
result = torch.allclose(x, y)
print(result)
Le résultat de sortie est:
True
Autres extensions