I’m looking for a method that takes an m-by-d tensor (x1
) and an n-by-d tensor (x2
) and computes the pairwise distance between each element of x2
with each elements of x1
.
I’ve seen nn.PairwiseDistance
in the documentation, which requires the tensors to have the same dimensions and only computes the distance between their corresponding rows.
An inefficient solution would be to use nn.PairwiseDistance
and iterate over different rows of x2
to compare each row of x2
with all rows of x1
.