Función PyTorch torch.arctan2

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.arctan2es una función en PyTorch para calcular el arco tangente bidimensional (arctangent2) elemento por elemento, contorch.atan2la misma funcionalidad.

Definición de la función

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

Descripción de parámetros

  • input: tensor numerador (coordenada y)
  • other: tensor denominador (coordenada x)
  • out: tensor de salida (opcional)

Ejemplo de uso

Ejemplo

import torch

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

# Calcular el arco tangente bidimensional
result = torch.arctan2(y, x)
print(result)

El resultado de salida es:

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

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones