PyTorch torch.isclose-Funktion
PyTorch torch Referenzhandbuch
torch.iscloseEs ist eine Funktion in PyTorch, mit der überprüft wird, ob zwei Tensoren nahe beieinanderliegen. Gibt einen booleschen Tensor zurück, der angibt, ob die entsprechenden Elemente in den beiden Tensoren innerhalb einer gegebenen Toleranz nahe beieinanderliegen.
Funktionsdefinition
torch.isclose(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])
# Überprüfen, ob die entsprechenden Elemente nahe beieinanderliegen
result = torch.isclose(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])
# Überprüfen, ob die entsprechenden Elemente nahe beieinanderliegen
result = torch.isclose(x, y)
print(result)
Das Ausgabeergebnis ist:
tensor([True, True, True])
Andere Erweiterungen