Fonction torch.arctan2 de PyTorch

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

torch.arctan2est une fonction de PyTorch utilisée pour calculer l'arctangente à deux arguments élément par élément (arctangente2), avec les mêmestorch.atan2fonctionnalité.

Définition de fonction

torch.arctan2(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 (optionnel)

Exemple d'utilisation

Exemple

import torch

# Créer des tenseurs
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 à deux arguments
result = torch.arctan2(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