Función PyTorch torch.isnan
Manual de referencia de Pytorch torch
torch.isnanEs una función en PyTorch utilizada para verificar si un valor es NaN.
Definición de la función
torch.isnan(input)
Ejemplo de uso
Ejemplo
import torch
import math
x = torch.tensor([1.0, math.nan, 3.0, float('nan')])
result = torch.isnan(x)
print(result)
import math
x = torch.tensor([1.0, math.nan, 3.0, float('nan')])
result = torch.isnan(x)
print(result)
El resultado de salida es:
tensor([False, True, False, True])
Otras extensiones