Función torch.isneginf de PyTorch
Manual de referencia de Pytorch torch
torch.isneginfEs una función de PyTorch utilizada para comprobar si es infinito negativo. Devuelve un tensor booleano, donde cada elemento es True si es infinito negativo.
Definición de la función
torch.isneginf(input)
Ejemplo de uso
Ejemplo
import torch
# Crear un tensor que contiene diferentes valores
x = torch.tensor([1.0, float('inf'), float('-inf'), 2.0])
# Comprobar si es infinito negativo
result = torch.isneginf(x)
print(result)
# Crear un tensor que contiene diferentes valores
x = torch.tensor([1.0, float('inf'), float('-inf'), 2.0])
# Comprobar si es infinito negativo
result = torch.isneginf(x)
print(result)
El resultado de salida es:
tensor([False, False, True, False])
Otras extensiones