Función PyTorch torch.atan2

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

torch.atan2Es la función en PyTorch que se utiliza para calcular la arcotangente bidimensional (arctangent2) elemento por elemento. Calcula el ángulo basándose en los dos valores de coordenadas (y, x), y el valor de retorno está en radianes, en el rango de [-pi, pi].

Definición de la función

torch.atan2(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 tensores
y = torch.tensor([0.0, 1.0, -1.0, 1.0])
x = torch.tensor([1.0, 0.0, 0.0, -1.0])

# Calcular la arcotangente bidimensional
result = torch.atan2(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