ott.geometry.costs.Bures.all_pairs_pairwise#

Bures.all_pairs_pairwise(x, y)#

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

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

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

Return type

ndarray

Returns

[num_a, num_b] matrix of pairwise cost evaluations.