Fonction torch.isreal de PyTorch

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

torch.isrealC'est une fonction dans PyTorch utilisée pour vérifier si une valeur est un nombre réel. Elle renvoie un tenseur booléen où, pour chaque élément, la valeur vaut True s'il s'agit d'un nombre réel (et non d'un nombre complexe).

Définition de fonction

torch.isreal(input)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur contenant des nombres réels et complexes
x = torch.tensor([1.0, 2+3j, 4.0, 5+1j])

# Vérifier s'il s'agit d'un nombre réel
result = torch.isreal(x)
print(result)

Le résultat de sortie est:

tensor([True, False, True, False])

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

Autres extensions