Función PyTorch torch.view_as_complex

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

torch.view_as_complexEs una función en PyTorch que se utiliza para convertir una vista de tensor real a un tensor complejo. Convierte la última dimensión de dos valores reales (parte real e imaginaria) a un único valor complejo.

Definición de la función

torch.view_as_complex(input)

Ejemplo de uso

Ejemplo

import torch

# Crear un tensor real, la última dimensión debe tener 2 elementos
x = torch.randn(3, 2)

# Convertir a vista compleja
y = torch.view_as_complex(x)

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

El resultado de salida es:

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

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

Otras extensiones