¿Cómo construir un tensor complejo con las partes reales e imaginarias dadas en PyTorch?

En este artículo, veremos cómo construir un tensor complejo con las partes reales e imaginarias dadas en PyTorch .

Ahora veremos cómo construir un número complejo en pytorch usando el método torch.complex() con la entrada dada como números reales e imaginarios.

Sintaxis: torch.complex(real, imag)

Parámetro:

• real: La parte real del tensor complejo. Debe ser flotante o doble.
• imag: La parte imaginaria del tensor complejo. Debe ser del mismo tipo que el real.

Devuelve: número complejo, si las entradas son torch.float32, debe ser torch.complex64. Si las entradas son torch.float64, debe ser torch.complex128.

Ejemplo 1:

En este ejemplo, primero creamos dos variables que son de tipo real y otra es de tipo flotante imaginario con la ayuda de torch.tensor y lo convertiremos en un número complejo usando el método torch.comple . Aquí, real es el número real e imag es el número imaginario, de modo que ambos pueden ser solo de tipo flotante o de tipo imaginario. Finalmente, mostraremos el tipo de datos del número complejo.

# import the torch module
import torch
 
# create real and img with float type
real = torch.tensor([78.2, 23.2], dtype=torch.float32)
img = torch.tensor([32, 41], dtype=torch.float32)
 
# display
print(real)
print(img)
 
# display  the complex number
print(torch.complex(real, img))
 
# display  the datatype of complex number
print(torch.complex(real, img).dtype)

Producción:

tensor([78.2000, 23.2000])

tensor([32., 41.])

tensor([78.2000+32.j, 23.2000+41.j])

antorcha.complex64

Ejemplo 2:

En este ejemplo, estamos creando 2 partes reales e imaginarias del tipo dtype=torch.double y construimos una salida compleja. Finalmente, estamos mostrando el tipo de datos del número complejo que es complejo 128.

# import the torch module
import torch
 
# create real and img with double type
real = torch.tensor([78, 23], dtype=torch.double)
img = torch.tensor([32, 41], dtype=torch.double)
 
# display
print(real)
print(img)
 
# display  the complex number
print(torch.complex(real, img))
 
# display  the datatype of complex number
print(torch.complex(real, img).dtype)

Producción:

tensor([78., 23.], dtype=torch.float64)

tensor([32., 41.], dtype=torch.float64)

tensor([78.+32.j, 23.+41.j], dtype=torch.complex128)

antorcha.complejo128

Ejemplo 3:

En este ejemplo, estamos creando 3 tipos flotantes reales con dype=torch.float64 , 3 tipos dobles con dtype=torch.double y construimos un número complejo con la ayuda de torch.complex.

# import the torch module
import torch
 
# create real and img with double type
real = torch.tensor([78, 23, 45], dtype=torch.float64)
img = torch.tensor([32, 41, 9], dtype=torch.double)
 
# display
print(real)
print(img)
 
# display  the complex number
print(torch.complex(real, img))
 
# display  the datatype of complex number
print(torch.complex(real, img).dtype)

Salida :

tensor([78., 23., 45.], dtype=torch.float64)

tensor([32., 41., 9.], dtype=torch.float64)

tensor([78.+32.j, 23.+41.j, 45.+9.j], dtype=torch.complex128)

antorcha.complejo128

        [78., 23., 45.]], dtype=torch.float64)

tensor([[32., 41., 9.],

        [78., 23., 45.]], dtype=torch.float64)

tensor([[78.+32.j, 23.+41.j, 45.+9.j],

        [78.+78.j, 23.+23.j, 45.+45.j]], dtype=torch.complex128)

antorcha.complejo128