ott.geometry.costs.UnbalancedBures.all_pairs_pairwise#

UnbalancedBures.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.