PyTorch torch.isinf función
Pytorch torch manual de referencia
torch.isinfEs una función en PyTorch utilizada para comprobar si un valor es infinito. Devuelve un tensor booleano, donde para cada elemento, si es infinito (infinito positivo o infinito negativo) es True.
Definición de la función
torch.isinf(input)
Ejemplo de uso
Ejemplo
import torch
# Crea un tensor que contiene diferentes valores
x = torch.tensor([1.0, float('inf'), float('-inf'), 2.0])
# Comprueba si es un valor infinito
result = torch.isinf(x)
print(result)
# Crea un tensor que contiene diferentes valores
x = torch.tensor([1.0, float('inf'), float('-inf'), 2.0])
# Comprueba si es un valor infinito
result = torch.isinf(x)
print(result)
El resultado de salida es:
tensor([False, True, True, False])
Otras extensiones