PyTorch torch.atan2 Funktion
PyTorch torch-Referenzhandbuch
torch.atan2Es ist eine Funktion in PyTorch, die verwendet wird, um den elementweisen zweidimensionalen Arkustangens (arctangent2) zu berechnen. Sie berechnet den Winkel basierend auf zwei Koordinatenwerten (y, x), und der Rückgabewert ist im Bogenmaß und liegt zwischen [-pi, pi].
Funktionsdefinition
torch.atan2(input, other, out=None)
Parameterbeschreibung
input: Zähler-Tensor (y-Koordinate)other: Nenner-Tensor (x-Koordinate)out: Ausgabe-Tensor (optional)
Anwendungsbeispiel
Beispiel
import torch
# Tensor erstellen
y = torch.tensor([0.0, 1.0, -1.0, 1.0])
x = torch.tensor([1.0, 0.0, 0.0, -1.0])
# Zweidimensionalen Arkustangens berechnen
result = torch.atan2(y, x)
print(result)
# Tensor erstellen
y = torch.tensor([0.0, 1.0, -1.0, 1.0])
x = torch.tensor([1.0, 0.0, 0.0, -1.0])
# Zweidimensionalen Arkustangens berechnen
result = torch.atan2(y, x)
print(result)
Das Ausgabeergebnis ist:
tensor([ 0.0000, 1.5708, -1.5708, 2.3562])
Andere Erweiterungen