Fonction torch.isfinite de PyTorch
Manuel de référence PyTorch torch
torch.isfiniteIl s'agit d'une fonction de PyTorch utilisée pour vérifier si les éléments d'un tenseur sont des valeurs finies. Elle renvoie un tenseur booléen, où pour chaque élément, il vaut True si la valeur est finie (pas l'infini ni NaN).
Définition de fonction
torch.isfinite(input)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur contenant différentes valeurs
x = torch.tensor([1.0, float('inf'), float('nan'), 2.0])
# Vérifier les valeurs finies
result = torch.isfinite(x)
print(result)
# Créer un tenseur contenant différentes valeurs
x = torch.tensor([1.0, float('inf'), float('nan'), 2.0])
# Vérifier les valeurs finies
result = torch.isfinite(x)
print(result)
La sortie est :
tensor([True, False, False, True])
Autres extensions