Fonction PyTorch torch.nan_to_num
Manuel de référence Pytorch torch
torch.nan_to_numIl s'agit d'une fonction de PyTorch utilisée pour remplacer les valeurs NaN dans les tenseurs par une valeur spécifiée. Elle peut également remplacer les valeurs infinies.
Définition de fonction
torch.nan_to_num(input, nan=0.0, posinf=None, neginf=None, out=None)
Exemples d'utilisation
Exemple
import torch
# Remplacer NaN par 0
x = torch.tensor([1.0, float('nan'), 3.0, float('inf'), float('-inf')])
résultat = torch.nan_to_num(x, nan=0.0)
print(result)
# Remplacer NaN par 0
x = torch.tensor([1.0, float('nan'), 3.0, float('inf'), float('-inf')])
résultat = torch.nan_to_num(x, nan=0.0)
print(result)
Autres extensions