Función torch.isclose de PyTorch
Manual de referencia de Pytorch torch
torch.iscloseEs una función en PyTorch utilizada para comprobar si dos tensores están próximos. Devuelve un tensor booleano que indica si los elementos correspondientes de los dos tensores están próximos dentro de un margen de tolerancia dado.
Definición de la función
torch.isclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False)
Ejemplo de uso
Ejemplo
import torch
# Crear dos tensores similares
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([1.0, 2.0, 3.0 + 1e-9])
# Comprobar si los elementos correspondientes son similares
result = torch.isclose(x, y)
print(result)
# Crear dos tensores similares
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([1.0, 2.0, 3.0 + 1e-9])
# Comprobar si los elementos correspondientes son similares
result = torch.isclose(x, y)
print(result)
El resultado de salida es:
tensor([True, True, True])
Otras extensiones