uniport.model.loss.distance_matrix

uniport.model.loss.distance_matrix(pts_src: Tensor, pts_dst: Tensor, p: int = 2)[source]

Returns the matrix of ||x_i-y_j||_p^p.

Parameters:
  • pts_src – [R, D] matrix

  • pts_dst – C, D] matrix

  • p – p-norm

Returns:

distance matrix

Return type:

[R, C] matrix