¿Cómo comparar dos tensores en PyTorch?

En este artículo vamos a ver cómo podemos comparar dos tensores en Pytorch. 

Podemos comparar dos tensores usando el método torch.eq(). Este método compara los elementos correspondientes de los tensores. Tiene que devolver rue en cada ubicación donde ambos tensores tienen el mismo valor; de lo contrario, devolverá falso.

función antorcha.eq():

Sintaxis: torch.eq( First_tensor, Second_tensor, out=None )

Parámetros: torch.eq() acepta como parámetros los tensores que queremos comparar.

Retorno: Devuelve un valor booleano. verdadero si los tensores son iguales, de lo contrario devolverá falso.

Ejemplo 1:

En este ejemplo, estamos comparando dos tensores 1-D usando la función torch.eq() en el lenguaje de programación python.

# import library
import torch
  
# Create first tensor
first = torch.Tensor([4.4, 2.4, -9.1, 
                      -5.31, 5.3])
  
# Create second tensor
second = torch.Tensor([4.4, 5.5, -9.1,
                       -5.31, 43])
  
# print first tensors
print("First Tensor:", first)
  
# print first tensors
print("Second Tensor:", second)
  
# Compare element wise tensors
# first and second
print(torch.eq(first, second))

Producción:

Ejemplo 2: 

Bajo este ejemplo, estamos comparando los tensores 2D PyTorch usando la función torch.eq().

# import library
import torch
  
# create two 2D tensors
first = torch.Tensor([[7, -2, 3],
                      [29, 9, -5],
                      [2, -8, 34],
                      [24, 62, 98]])
  
second = torch.Tensor([[7, -5, 3],
                       [26, 9, -4],
                       [3, -8, 43],
                       [23, -62, 98]])
  
# print first tensors
print("First Tensor:", first)
  
# print second tensors
print("Second Tensor:\n", second)
  
  
print("After Comparing Both Tensors")
  
# Compare element wise tensors first
# and second
print(torch.eq(first, second))

Producción: