Función torch.eq de PyTorch
Manual de referencia de Pytorch torch
torch.eqEs una función de PyTorch utilizada para comparar la igualdad elemento por elemento.
Definición de la función
torch.eq(input, other, out)
Ejemplo de uso
Ejemplo
import torch
a = torch.tensor([1, 2, 3, 4])
b = torch.tensor([1, 2, 0, 4])
result = torch.eq(a, b)
print(result)
a = torch.tensor([1, 2, 3, 4])
b = torch.tensor([1, 2, 0, 4])
result = torch.eq(a, b)
print(result)
El resultado de salida es:
tensor([True, True, False, True])
Otras extensiones