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.