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#

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

Solve quadratic regularized OT problem.

gromov_wasserstein.GromovWasserstein(*args)

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

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

Holds the output of the Gromov-Wasserstein solver.

Barycenter Solvers#

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

State of the GW barycenter problem.

gw_barycenter.GromovWassersteinBarycenter([...])

Gromov-Wasserstein barycenter solver.