ott.initializers.linear package#

Sinkhorn Initializers#

initializers.DefaultInitializer()

Default initialization of Sinkhorn dual potentials/primal scalings.

initializers.GaussianInitializer()

Gaussian initializer [Thornton and Cuturi, 2022].

initializers.SinkhornInitializer()

Base class for Sinkhorn initializers.

Low-rank Sinkhorn 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].