ott.solvers.quadratic package#

Gromov-Wasserstein Solvers#

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

Solve quadratic regularized OT problem.

gromov_wasserstein.GromovWasserstein(*args)

Gromov-Wasserstein solver.

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

Holds the output of the Gromov-Wasserstein solver.

Barycenter Solvers#

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

Holds the state of the GWBarycenterProblem.

gw_barycenter.GromovWassersteinBarycenter([...])

Gromov-Wasserstein barycenter solver of the GWBarycenterProblem.