Función torch.equal de PyTorch
Manual de referencia de Pytorch torch
torch.equalEs una función en PyTorch que se usa para comprobar si dos tensores son iguales. Si los dos tensores tienen la misma forma y elementos, devuelve True.
Definición de la función
torch.equal(input, other)
Ejemplo de uso
Ejemplo
import torch
# Crea dos tensores idénticos
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 2, 3])
# Comprueba si son iguales
result = torch.equal(x, y)
print(result)
# Crea tensores diferentes
a = torch.tensor([1, 2, 4])
b = torch.tensor([1, 2, 3])
result2 = torch.equal(a, b)
print(result2)
# Crea dos tensores idénticos
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 2, 3])
# Comprueba si son iguales
result = torch.equal(x, y)
print(result)
# Crea tensores diferentes
a = torch.tensor([1, 2, 4])
b = torch.tensor([1, 2, 3])
result2 = torch.equal(a, b)
print(result2)
El resultado de salida es:
True False
Otras extensiones