ott.problems.linear package#

OT Problems#

linear_problem.LinearProblem(geom[, a, b, ...])

Linear OT problem.

barycenter_problem.BarycenterProblem(y[, b, ...])

Wasserstein barycenter problem [Cuturi and Doucet, 2014].

Dual Potentials#

potentials.DualPotentials(f, g, *, cost_fn)

The Kantorovich dual potential functions \(f\) and \(g\).

potentials.EntropicPotentials(f_xy, g_xy, prob)

Dual potential functions from finite samples [Pooladian and Niles-Weed, 2021].