ott.problems.linear.potentials.DualPotentials.distance#

DualPotentials.distance(src, tgt)[source]#

Evaluate 2-Wasserstein distance between samples using dual potentials.

Uses Eq. 5 from [Makkuva et al., 2020] when given in corr form, direct estimation by integrating dual function against points when using dual form.

Parameters:

  • src (Array) – Samples from the source distribution, array of shape [n, d].

  • tgt (Array) – Samples from the target distribution, array of shape [m, d].

Return type:

float

Returns:

Wasserstein distance.