Fonction torch.view_as_complex de PyTorch

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

torch.view_as_complexIl s'agit d'une fonction dans PyTorch utilisée pour convertir une vue de tenseur réel en un tenseur complexe. Elle convertit la dernière dimension à partir de deux valeurs réelles (partie réelle et partie imaginaire) en une seule valeur complexe.

Définition de fonction

torch.view_as_complex(input)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur réel ; la dernière dimension doit avoir 2 éléments
x = torch.randn(3, 2)

# Convertir en vue complexe
y = torch.view_as_complex(x)

print("Forme du tenseur réel d'origine :", x.shape)
print("Forme du tenseur complexe :", y.shape)
print("Tenseur complexe :", y)

Le résultat de sortie est:

原始实数张量形状: torch.Size([3, 2])
复数张量形状: torch.Size([3])
复数张量: tensor([ 1.2345+0.5678j, -0.3456+1.2345j,  2.3456-0.9876j])

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

Autres extensions