ott.geometry.costs.CostFn.all_pairs#

CostFn.all_pairs(x, y)[source]#

Compute matrix of all costs (including 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 cost evaluations.