PyTorch torch.complex Funktion


Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.complexist eine Funktion in PyTorch, die verwendet wird, um komplexe Tensoren aus Real- und Imaginärteil zu erstellen.

Funktionsdefinition

torch.complex(real, imag)

Anwendungsbeispiel

Beispiel

import torch

# Tensor für Realteil erstellen
real = torch.tensor([1.0, 2.0, 3.0])

# Tensor für Imaginärteil erstellen
imag = torch.tensor([4.0, 5.0, 6.0])

# Komplexen Tensor aus Real- und Imaginärteilen erstellen
x = torch.complex(real, imag)

print(x)
print("Realteil:", x.real)
print("Imaginärteil:", x.imag)

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen