Fonction torch.isclose de PyTorch
Manuel de référence Pytorch torch
torch.iscloseIl s'agit d'une fonction de PyTorch utilisée pour vérifier si deux tenseurs sont proches. Renvoie un tenseur booléen indiquant si les éléments correspondants des deux tenseurs sont proches à une tolérance donnée.
Définition de fonction
torch.isclose(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 les éléments correspondants sont proches
result = torch.isclose(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 les éléments correspondants sont proches
result = torch.isclose(x, y)
print(result)
Le résultat de sortie est:
tensor([True, True, True])
Autres extensions