Función torch.view_as_real de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.view_as_realEs una función en PyTorch utilizada para convertir la vista de un tensor complejo en un tensor real. Divide cada valor complejo (parte real e imaginaria) en dos valores separados, y la última dimensión tendrá dos elementos.

Definición de la función

torch.view_as_real(input)

Ejemplo de uso

Ejemplo

import torch

# Crear un tensor complejo
x = torch.tensor([1+2j, 3+4j, 5+6j])

# Convertir a vista real
y = torch.view_as_real(x)

print("Forma del tensor complejo original:", x.shape)
print("Forma del tensor real:", y.shape)
print("Tensor real:")
print(y)

El resultado de salida es:

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

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones