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.