ott.problems.quadratic

The ott.problems.quadratic module describes the quadratic assignment problem and its generalizations, including notably the fused-problem (including a linear term) and the more advanced GW barycenter problem.

OT Problems

quadratic_problem.QuadraticProblem(geom_xx, ...)	Quadratic OT problem.
gw_barycenter.GWBarycenterProblem([y, b, ...])	(Fused) Gromov-Wasserstein barycenter problem [Peyré et al., 2016, Titouan et al., 2019].

Costs

quadratic_costs.GWLoss(f1, f2, h1, h2)	Efficient decomposition of the Gromov-Wasserstein loss function.
quadratic_costs.make_square_loss()	Squared Euclidean loss for Gromov-Wasserstein.
quadratic_costs.make_kl_loss([clipping_value])	Kullback-Leibler loss for Gromov-Wasserstein.