Fonction PyTorch torch.complex

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.complexest une fonction de PyTorch utilisée pour créer des tenseurs complexes à partir de parties réelles et imaginaires.

Définition de fonction

torch.complex(real, imag)

Exemple d'utilisation

Exemple

import torch

# Créer le tenseur de partie réelle
real = torch.tensor([1.0, 2.0, 3.0])

# Créer le tenseur de partie imaginaire
imag = torch.tensor([4.0, 5.0, 6.0])

# 从实部和虚部创建复数张量
x = torch.complex(real, imag)

print(x)
print("Real part:", x.real)
print("Imaginary part:", x.imag)

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions