Fonction torch.atan2 de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.atan2Il s'agit d'une fonction de PyTorch utilisée pour calculer l'arctangente bidimensionnelle élément par élément (arctangent2). Elle calcule l'angle à partir de deux valeurs de coordonnées (y, x), et la valeur renvoyée est en radians, comprise entre [-pi, pi].

Définition de fonction

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

Description du paramètre

  • input: Tenseur numérateur (coordonnée y)
  • other: Tenseur dénominateur (coordonnée x)
  • out: Tenseur de sortie (facultatif)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur
y = torch.tensor([0.0, 1.0, -1.0, 1.0])
x = torch.tensor([1.0, 0.0, 0.0, -1.0])

# Calculer l'arctangente bidimensionnelle
result = torch.atan2(y, x)
print(result)

Le résultat de sortie est:

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

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions