¿Cómo calcular el ángulo del elemento del tensor de entrada dado en PyTorch?

En este artículo, veremos cómo calcular el ángulo elemento de un tensor de entrada dado en PyTorch .

método antorcha.ángulo()

Pytorch es un marco de aprendizaje profundo de código abierto disponible con una interfaz de Python y C++. Pytorch reside dentro del módulo de la antorcha. En PyTorch, usaremos el método torch.angle() para calcular el ángulo del elemento, la entrada se pasa en forma de tensor en radianes. Para calcular el ángulo en grados, multiplicamos el ángulo en radianes por 180/NumPy.pi, antes de continuar, veamos la sintaxis del método dado:

Sintaxis: torch.angle(tens, *, out=Ninguno)

Parámetros:

• decenas: este es nuestro tensor de entrada.
• out – Este es nuestro tensor de salida (opcional). 

Devuelve: Devuelve el ángulo de los elementos del tensor dado.

Ejemplo 1:

En este ejemplo, calcularemos ángulos por elementos en radianes de un tensor 1-D dado.

# Import the required libraries
import torch
  
# define a complex tensor
input = torch.tensor([1 - 2j, 2 + 1j, 3 - 2j, -4 + 3j, -5 - 2j])
  
# print the above define tensor
print("\n Input Tensor: ", input)
  
# compute the elements-wise angle in radians
ang = torch.angle(input)
  
# print the computed elements-wise angle in radians
print("\n Elements-wise angles in radian: ", ang)

Producción:

 

Ejemplo 2:

En este ejemplo, calcularemos ángulos por elementos para convertir el ángulo en un grado del tensor 1-D dado.

# Import the required libraries
import torch
from numpy import pi
  
# define a complex tensor
input = torch.tensor([1 - 2j, 2 + 1j, 3 - 2j, -4 + 3j, -5 - 2j])
  
# print the above define tensor
print("\n\nInput Tensor: ", input)
  
# compute the elements-wise angle in radians
ang = torch.angle(input)
  
# convert the angle in degree
deg = ang*180/pi
  
# print the computed elements-wise angle in degree
print("\nElements-wise angles in degree: ", deg)

Producción:

 

Ejemplo 3:

En este ejemplo, calcularemos ángulos por elementos para convertir el ángulo en un grado del tensor 2-D dado.

# Import the required libraries
import torch
from numpy import pi
  
# define a complex tensor
input = torch.tensor([[1 - 2j, 2 + 3j, 3 - 3j],
                      [4 + 3j, 5 - 4j, -6 + 2j],
                      [-7 - 2j, 8 + 2j, 9 - 4j]])
  
# print the above define tensor
print("\n Input Tensor:\n ", input)
  
# compute the elements-wise angle in radians
radians = torch.angle(input)
  
# print the computed elements-wise angle in radians
print("\n Elements-wise angles in radians:\n ", radians)
  
# convert the angle in degree
degree = radians * 180/pi
  
# print the computed elements-wise angle in degree
print("\n Elements-wise angles in degree:\n ", degree)

Producción: