Función torch.allclose de PyTorch
Manual de referencia de Pytorch torch
torch.allcloseEs una función en PyTorch que se utiliza para comprobar si todos los elementos de dos tensores están cerca. Se usa para determinar si dos tensores son iguales dentro de una tolerancia dada.
Definición de la función
torch.allclose(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])
# Verificar si todos los elementos están cerca
result = torch.allclose(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])
# Verificar si todos los elementos están cerca
result = torch.allclose(x, y)
print(result)
El resultado de salida es:
True
Otras extensiones