Fonction torch.imag de PyTorch

Pytorch torch 参考手册 Manuel de référence Pytorch torch

torch.imagest une fonction de PyTorch utilisée pour obtenir la partie imaginaire d’un tenseur complexe.

Définition de fonction

torch.imag(input)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur complexe
x = torch.complex(torch.tensor([1.0, 2.0]), torch.tensor([3.0, 4.0]))

# Obtenir la partie imaginaire
y = torch.imag(x)
print("Partie imaginaire :", y)

Le résultat de sortie est:

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

Pytorch torch 参考手册 Manuel de référence Pytorch torch

Autres extensions