ott.core.initializers.GaussianInitializer#

class ott.core.initializers.GaussianInitializer[source]#

Gaussian initializer [Thornton and Cuturi, 2022].

Compute Gaussian approximations of each point cloud, then compute closed from Kantorovich potential between Gaussian approximations using Brenier’s theorem (adapt convex/Brenier potential to Kantorovich). Use this Gaussian potential to initialize Sinkhorn potentials/scalings.

Methods

init_dual_a(ot_prob, lse_mode)

Gaussian initialization function.

init_dual_b(ot_prob, lse_mode)

Initialize Sinkhorn potential/scaling g_v.