Cómo calcular la distancia por pares entre dos vectores en PyTorch

En este artículo, discutiremos cómo calcular la distancia por pares entre dos vectores en PyTorch . 

El tamaño del vector debe ser el mismo y podemos usar el método PairwiseDistance() para calcular la distancia por pares entre dos vectores.

Método PairwiseDistance()

El método PairwiseDistance() calcula la distancia por pares entre dos vectores utilizando la norma p. Este método lo proporciona el módulo de la antorcha. La siguiente sintaxis se utiliza para calcular la distancia por pares.

Sintaxis: torch.nn.PairwiseDistance(p=2)

Devolver: este método devuelve la distancia por pares entre dos vectores.

Ejemplo 1:

El siguiente programa es para entender cómo calcular la distancia por pares entre dos vectores.

import torch
  
# define first vector
vec1 = torch.tensor([1., 2., 3., 4.])
  
# define second vector
vec2 = torch.tensor([5., 6., 7., 8.])
  
# display tensors
print("\n First Vector: ", vec1)
print("\n Second Vector: ", vec2)
  
# define an instance of the PairwiseDistance
pdist = torch.nn.PairwiseDistance(p=2)
  
# compute the pairwise distance
result = pdist(vec1, vec2)
  
# display result
print("\n Pairwise Distance:", result)

Producción:

 

Ejemplo 2:

import torch
  
# define first vector
vec1 = torch.tensor([[0.2245, 0.2959, 0.3597, 0.6766],
                     [0.3685, 0.6469, 0.3765, 0.7898],
                     [0.4577, 0.3228, 0.4699, 0.2389]])
  
# define second vector
vec2 = torch.tensor([[0.2423, 0.4667, 0.4434, 0.3598],
                     [0.2956, 0.6932, 0.5387, 0.2245],
                     [0.8277, 0.2597, 0.9834, 0.9987]])
  
# display tensors
print("\n First Vector: \n", vec1)
print("\n Second Vector: \n", vec2)
  
# define an instance of the PairwiseDistance
pdist = torch.nn.PairwiseDistance(p=2)
  
# compute the pairwise distance
result = pdist(vec1, vec2)
  
# display result
print("\n Pairwise Distance: \n", result)

Producción: