Fonction torch.view_as_real de PyTorch

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

torch.view_as_realC'est une fonction de PyTorch utilisée pour convertir une vue de tenseur complexe en un tenseur réel. Elle divise chaque valeur complexe (parties réelle et imaginaire) en deux valeurs distinctes, et la dernière dimension aura deux éléments.

Définition de fonction

torch.view_as_real(input)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur complexe
x = torch.tensor([1+2j, 3+4j, 5+6j])

# Convertir en vue réelle
y = torch.view_as_real(x)

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

Le résultat de sortie est:

原始复数张量形状: torch.Size([3])
实数张量形状: torch.Size([3, 2])
实数张量:
tensor([[1., 2.],
        [3., 4.],
        [5., 6.]])

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

Autres extensions