Fonction torch.is_complex de PyTorch
Manuel de référence PyTorch torch
torch.is_complexIl s'agit d'une fonction de PyTorch utilisée pour vérifier si le type de données du tenseur est un type complexe.
Définition de fonction
torch.is_complex(input)
Exemple d'utilisation
Exemple
import torch
# Vérifier un tenseur réel
x = torch.tensor([1.0, 2.0, 3.0])
print("Réel :", torch.is_complex(x))
# Vérifier un tenseur complexe
y = torch.tensor([1+2j, 3+4j])
print("Complexe :", torch.is_complex(y))
# Vérifier un tenseur réel
x = torch.tensor([1.0, 2.0, 3.0])
print("Réel :", torch.is_complex(x))
# Vérifier un tenseur complexe
y = torch.tensor([1+2j, 3+4j])
print("Complexe :", torch.is_complex(y))
Le résultat de sortie est:
实数: False 复数: True
Autres extensions