ott.neural.methods.expectile_neural_dual.ENOTPotentials.distance

ott.neural.methods.expectile_neural_dual.ENOTPotentials.distance#

ENOTPotentials.distance(src, tgt)#

Evaluate Wasserstein distance between samples using dual potentials.

This uses direct estimation of potentials against measures when dual functions are provided in usual form. This expression is valid for any cost function.

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 using specified cost function.