ott.initializers.linear

Initializers for linear OT problems, focusing on Sinkhorn and low-rank solvers.

Sinkhorn Initializers

initializers.DefaultInitializer()	Default initialization of Sinkhorn dual potentials/primal scalings.
initializers.GaussianInitializer()	Gaussian initializer [Thornton and Cuturi, 2022].
initializers.SortingInitializer([...])	Sorting initializer [Thornton and Cuturi, 2022].
initializers.SubsampleInitializer(subsample_n_x)	Subsample initializer [Thornton and Cuturi, 2022].

Low-rank Sinkhorn Initializers

initializers_lr.LRInitializer(rank, **kwargs)	Base class for low-rank initializers.
initializers_lr.RandomInitializer(rank, **kwargs)	Low-rank Sinkhorn factorization using random factors.
initializers_lr.Rank2Initializer(rank, **kwargs)	Low-rank Sinkhorn factorization using rank-2 factors [Scetbon et al., 2021].
initializers_lr.KMeansInitializer(rank[, ...])	K-means initializer for low-rank Sinkhorn [Scetbon and Cuturi, 2022].
initializers_lr.GeneralizedKMeansInitializer(rank)	Generalized k-means initializer [Scetbon and Cuturi, 2022].