Fonction torch.isinf de PyTorch
Manuel de référence Pytorch torch
torch.isinfest une fonction de PyTorch utilisée pour vérifier si une valeur est infinie. Renvoie un tenseur booléen où, pour chaque élément, il vaut True si la valeur est infinie (infini positif ou infini négatif).
Définition de fonction
torch.isinf(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 si les valeurs sont infinies
result = torch.isinf(x)
print(result)
# Créer un tenseur contenant différentes valeurs
x = torch.tensor([1.0, float('inf'), float('-inf'), 2.0])
# Vérifier si les valeurs sont infinies
result = torch.isinf(x)
print(result)
Le résultat de sortie est:
tensor([False, True, True, False])
Autres extensions