ott.geometry.costs.SqPNorm.all_pairs_pairwise#

SqPNorm.all_pairs_pairwise(x, y)#

Compute matrix of all pairwise-costs (no norms) for vectors in x / y.

Parameters

  • x (Array) – [num_a, d] jnp.ndarray

  • y (Array) – [num_b, d] jnp.ndarray

Return type

Array

Returns

[num_a, num_b] matrix of pairwise cost evaluations.