Función torch.isfinite de PyTorch
Manual de referencia de PyTorch torch
torch.isfiniteEs una función de PyTorch que se utiliza para comprobar si los elementos de un tensor tienen valores finitos. Devuelve un tensor booleano que, para cada elemento, es True si el valor es finito (no es infinito ni NaN).
Definición de la función
torch.isfinite(input)
Ejemplo de uso
Ejemplo
import torch
# Crea un tensor con diferentes valores
x = torch.tensor([1.0, float('inf'), float('nan'), 2.0])
# Comprueba si hay valores finitos
result = torch.isfinite(x)
print(result)
# Crea un tensor con diferentes valores
x = torch.tensor([1.0, float('inf'), float('nan'), 2.0])
# Comprueba si hay valores finitos
result = torch.isfinite(x)
print(result)
El resultado de salida es:
tensor([True, False, False, True])
Otras extensiones