PyTorch torch.equal-Funktion
PyTorch torch Referenzhandbuch
torch.equalEs ist eine Funktion in PyTorch, mit der geprüft wird, ob zwei Tensoren gleich sind. Wenn die beiden Tensoren dieselbe Form und dieselben Elemente haben, gibt sie True zurück.
Funktionsdefinition
torch.equal(input, other)
Anwendungsbeispiel
Beispiel
import torch
# Zwei identische Tensoren erstellen
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 2, 3])
# Prüfen, ob sie gleich sind
result = torch.equal(x, y)
print(result)
# Unterschiedliche Tensoren erstellen
a = torch.tensor([1, 2, 4])
b = torch.tensor([1, 2, 3])
result2 = torch.equal(a, b)
print(result2)
# Zwei identische Tensoren erstellen
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 2, 3])
# Prüfen, ob sie gleich sind
result = torch.equal(x, y)
print(result)
# Unterschiedliche Tensoren erstellen
a = torch.tensor([1, 2, 4])
b = torch.tensor([1, 2, 3])
result2 = torch.equal(a, b)
print(result2)
Das Ausgabeergebnis ist:
True False
Andere Erweiterungen