PyTorch torch.arctan2-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.arctan2ist eine Funktion in PyTorch, die verwendet wird, um den elementweisen Arkustangens mit zwei Argumenten (arctangent2) zu berechnen, mit denselbentorch.atan2Funktionalität.

Funktionsdefinition

torch.arctan2(input, other, out=None)

Parameterbeschreibung

  • input: Zähler-Tensor (y-Koordinate)
  • other: Nenner-Tensor (x-Koordinate)
  • out: Ausgabe-Tensor (optional)

Anwendungsbeispiel

Beispiel

import torch

# Tensoren erstellen
y = torch.tensor([0.0, 1.0, -1.0, 1.0])
x = torch.tensor([1.0, 0.0, 0.0, -1.0])

# Arkustangens mit zwei Argumenten berechnen
Ergebnis = torch.arctan2(y, x)
print(result)

Das Ausgabeergebnis ist:

tensor([ 0.0000,  1.5708, -1.5708,  2.3562])

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen