In this article, we will discuss how to compute the pairwise distance between two vectors in PyTorch.
The vector size should be the same and we can use PairwiseDistance() method to compute the pairwise distance between two vectors.
PairwiseDistance() method
PairwiseDistance() method computes the pairwise distance between two vectors using the p-norm. This method is provided by the torch module. The below syntax is used to compute pairwise distance.
Syntax – torch.nn.PairwiseDistance(p=2)
Return – This method Returns the pairwise distance between two vectors.
Example 1:
The following program is to understand how to compute the pairwise distance between two vectors.
Python3
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) |
Output:
Example 2:
Python3
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) |
Output:
