Fonction torch.angle de PyTorch
Manuel de référence Pytorch torch
torch.angleC'est une fonction de PyTorch utilisée pour renvoyer l'angle de phase (argument) d'un tenseur complexe. La valeur renvoyée est en radians, avec une plage de [-pi, pi].
Définition de fonction
torch.angle(input, out=None)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur complexe
x = torch.tensor([1+1j, 1+0j, 0+1j, -1+0j])
# Renvoyer l'angle de phase
result = torch.angle(x)
print(result)
# Créer un tenseur complexe
x = torch.tensor([1+1j, 1+0j, 0+1j, -1+0j])
# Renvoyer l'angle de phase
result = torch.angle(x)
print(result)
Le résultat de sortie est:
tensor([ 0.7854, 0.0000, 1.5708, 3.1416])
Autres extensions