Función torch.complex de PyTorch

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

torch.complexEs la función en PyTorch que se utiliza para crear tensores complejos a partir de la parte real y la parte imaginaria.

Definición de la función

torch.complex(real, imag)

Ejemplo de uso

Ejemplo

import torch

# Crear el tensor de la parte real
real = torch.tensor([1.0, 2.0, 3.0])

# Crear el tensor de la parte imaginaria
imag = torch.tensor([4.0, 5.0, 6.0])

# Crear el tensor complejo a partir de la parte real e imaginaria
x = torch.complex(real, imag)

print(x)
print("Parte real:", x.real)
print("Parte imaginaria:", x.imag)

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

Otras extensiones