Fonction torch.isposinf de PyTorch
Manuel de référence Pytorch torch
torch.isposinfest une fonction dans PyTorch utilisée pour vérifier s'il s'agit de l'infini positif. Renvoie un tenseur booléen qui vaut True pour chaque élément s'il s'agit de l'infini positif.
Définition de fonction
torch.isposinf(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 positif
result = torch.isposinf(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 positif
result = torch.isposinf(x)
print(result)
Le résultat de sortie est:
tensor([False, True, False, False])
Autres extensions