Fonction torch.isneginf de PyTorch
Manuel de référence Pytorch torch
torch.isneginfC'est une fonction de PyTorch utilisée pour vérifier l'infini négatif. Renvoie un tenseur booléen où chaque élément est True s'il s'agit de l'infini négatif.
Définition de fonction
torch.isneginf(input)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur contenant différentes valeurs
x = torch.tensor([1.0, float('inf'), float('-inf'), 2.0])
# Vérifier s'il s'agit de l'infini négatif
result = torch.isneginf(x)
print(result)
# Créer un tenseur contenant différentes valeurs
x = torch.tensor([1.0, float('inf'), float('-inf'), 2.0])
# Vérifier s'il s'agit de l'infini négatif
result = torch.isneginf(x)
print(result)
Le résultat de sortie est:
tensor([False, False, True, False])
Autres extensions