ott.geometry.costs.SqPNorm.all_pairs

ott.geometry.costs.SqPNorm.all_pairs#

SqPNorm.all_pairs(x, y)#

Compute matrix of all pairwise costs, including the norms.

Parameters:
  • x (Array) – Array of shape [n, ...].

  • y (Array) – Array of shape [m, ...].

Return type:

Array

Returns:

Array of shape [n, m] of cost evaluations.