PyTorch torch.allclose Funktion
PyTorch torch-Referenzhandbuch
torch.allcloseEine Funktion in PyTorch, mit der überprüft wird, ob alle Elemente zweier Tensoren nahe beieinanderliegen. Sie wird verwendet, um festzustellen, ob zwei Tensoren innerhalb einer gegebenen Toleranz gleich sind.
Funktionsdefinition
torch.allclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False)
Anwendungsbeispiel
Beispiel
import torch
# Zwei ähnliche Tensoren erstellen
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([1.0, 2.0, 3.0 + 1e-9])
# Prüfen, ob alle Elemente nahezu gleich sind
result = torch.allclose(x, y)
print(result)
# Zwei ähnliche Tensoren erstellen
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([1.0, 2.0, 3.0 + 1e-9])
# Prüfen, ob alle Elemente nahezu gleich sind
result = torch.allclose(x, y)
print(result)
Das Ausgabeergebnis ist:
True
Andere Erweiterungen