Fonction torch.polar de PyTorch
Manuel de référence PyTorch torch
torch.polarest une fonction de PyTorch utilisée pour créer des tenseurs complexes à partir de coordonnées polaires.
Définition de fonction
torch.polar(abs, angle)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur de valeur absolue (magnitude)
abs_val = torch.tensor([1.0, 2.0, 3.0])
# Créer un tenseur d'angle (radian)
angle = torch.tensor([0.0, torch.pi/2, torch.pi])
# Créer un tenseur complexe à partir de coordonnées polaires
x = torch.polar(abs_val, angle)
print(x)
print("Valeur absolue :", x.abs())
print("Angle :", x.angle())
# Créer un tenseur de valeur absolue (magnitude)
abs_val = torch.tensor([1.0, 2.0, 3.0])
# Créer un tenseur d'angle (radian)
angle = torch.tensor([0.0, torch.pi/2, torch.pi])
# Créer un tenseur complexe à partir de coordonnées polaires
x = torch.polar(abs_val, angle)
print(x)
print("Valeur absolue :", x.abs())
print("Angle :", x.angle())
Autres extensions