ott.solvers.quadratic#

The ott.solvers.quadratic module holds the important family of Gromov-Wasserstein (GW) solvers and related GW barycenters. They are designed to solve ott.problems.quadratic problems.

Gromov-Wasserstein Solvers#

solve(geom_xx, geom_yy[, geom_xy, ...])

Solve quadratic regularized OT problem using a Gromov-Wasserstein solver.

gromov_wasserstein.GromovWasserstein(*args)

Gromov-Wasserstein solver [Peyré et al., 2016].

gromov_wasserstein.GWOutput([costs, ...])

Holds the output of the Gromov-Wasserstein solver.

gromov_wasserstein_lr.LRGromovWasserstein(rank)

Low-rank Gromov-Wasserstein solver [Scetbon et al., 2023].

gromov_wasserstein_lr.LRGWOutput(q, r, g, ...)

Transport interface for a low-rank GW solution.

lower_bound.LowerBoundSolver([epsilon, ...])

Lower bound OT solver.

Barycenter Solvers#

gw_barycenter.GWBarycenterState([cost, x, ...])

State of the GW barycenter problem.

gw_barycenter.GromovWassersteinBarycenter([...])

Gromov-Wasserstein barycenter solver.