Función torch.nan_to_num de PyTorch
Manual de referencia de Pytorch torch
torch.nan_to_numEs una función en PyTorch utilizada para reemplazar los valores NaN en tensores por un valor especificado. También puede reemplazar valores infinitos.
Definición de la función
torch.nan_to_num(input, nan=0.0, posinf=None, neginf=None, out=None)
Ejemplo de uso
Ejemplo
import torch
# Reemplazar NaN por 0
x = torch.tensor([1.0, float('nan'), 3.0, float('inf'), float('-inf')])
result = torch.nan_to_num(x, nan=0.0)
print(result)
# Reemplazar NaN por 0
x = torch.tensor([1.0, float('nan'), 3.0, float('inf'), float('-inf')])
result = torch.nan_to_num(x, nan=0.0)
print(result)
Otras extensiones