dgs.models.metric.metric.PairwiseDistanceMetric.forward

PairwiseDistanceMetric.forward(input1: torch.Tensor, input2: torch.Tensor) → torch.Tensor

Compute the pairwise distance between the two inputs, where the second dimension has to match.

Parameters:

• input1 – tensor of shape [a x E]

• input2 – tensor of shape [a x E], has to have the same shape as input1.

Returns:

tensor of shape [a (x 1)] containing the distances.