ott.geometry.costs.SqPNorm.all_pairs_pairwise
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.ndarrayy (
Array
) – [num_b, d] jnp.ndarray
- Return type
Array
- Returns
[num_a, num_b] matrix of pairwise cost evaluations.