Función torch.imag de PyTorch

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

torch.imagEs una función en PyTorch que se utiliza para obtener la parte imaginaria (imaginary part) de un tensor complejo.

Definición de la función

torch.imag(input)

Ejemplo de uso

Ejemplo

import torch

# Crear un tensor complejo
x = torch.complex(torch.tensor([1.0, 2.0]), torch.tensor([3.0, 4.0]))

# Obtener la parte imaginaria
y = torch.imag(x)
print("Parte imaginaria:", y)

El resultado de salida es:

虚部: tensor([3., 4.])

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

Otras extensiones