PyTorch torch.view_as_complex-Funktion
PyTorch torch Referenzhandbuch
torch.view_as_complexEs ist eine Funktion in PyTorch, die verwendet wird, um eine reelle Tensoransicht in einen komplexen Tensor umzuwandeln. Sie wandelt die letzte Dimension von zwei reellen Werten (Realteil und Imaginärteil) in einen einzelnen komplexen Wert um.
Funktionsdefinition
torch.view_as_complex(input)
Anwendungsbeispiel
Beispiel
import torch
# Erstellen Sie einen reellen Tensor; die letzte Dimension muss 2 Elemente haben
x = torch.randn(3, 2)
# In komplexe Ansicht umwandeln
y = torch.view_as_complex(x)
print("Ursprüngliche Form des reellen Tensors:", x.shape)
print("Form des komplexen Tensors:", y.shape)
print("Komplexer Tensor:", y)
# Erstellen Sie einen reellen Tensor; die letzte Dimension muss 2 Elemente haben
x = torch.randn(3, 2)
# In komplexe Ansicht umwandeln
y = torch.view_as_complex(x)
print("Ursprüngliche Form des reellen Tensors:", x.shape)
print("Form des komplexen Tensors:", y.shape)
print("Komplexer Tensor:", y)
Das Ausgabeergebnis ist:
原始实数张量形状: torch.Size([3, 2]) 复数张量形状: torch.Size([3]) 复数张量: tensor([ 1.2345+0.5678j, -0.3456+1.2345j, 2.3456-0.9876j])
Andere Erweiterungen