ott.initializers.linear.initializers.GaussianInitializer

ott.initializers.linear.initializers.GaussianInitializer#

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

Gaussian initializer [Thornton and Cuturi, 2022].

Compute Gaussian approximations of each PointCloud, 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[, rng])

Initialize Sinkhorn potential/scaling f_u.

init_dual_b(ot_prob, lse_mode[, rng])

Initialize Sinkhorn potential/scaling g_v.