Función torch.angle de PyTorch
Manual de referencia de PyTorch torch
torch.angleEs una función en PyTorch que se utiliza para devolver el ángulo de fase (argumento) de tensores complejos. El valor de retorno está en radianes, con un rango entre [-pi, pi].
Definición de la función
torch.angle(input, out=None)
Ejemplo de uso
Ejemplo
import torch
# Crear un tensor complejo
x = torch.tensor([1+1j, 1+0j, 0+1j, -1+0j])
# Devolver el ángulo de fase
result = torch.angle(x)
print(result)
# Crear un tensor complejo
x = torch.tensor([1+1j, 1+0j, 0+1j, -1+0j])
# Devolver el ángulo de fase
result = torch.angle(x)
print(result)
El resultado de salida es:
tensor([ 0.7854, 0.0000, 1.5708, 3.1416])
Otras extensiones