ott.initializers.linear package
Contents
ott.initializers.linear package#
Sinkhorn Initializers#
Default initialization of Sinkhorn dual potentials/primal scalings. |
|
Gaussian initializer [Thornton and Cuturi, 2022]. |
|
Base class for Sinkhorn initializers. |
Low-rank Sinkhorn Initializers#
|
Low-rank Sinkhorn factorization using random factors. |
|
Low-rank Sinkhorn factorization using rank-2 factors [Scetbon et al., 2021]. |
|
K-means initializer for low-rank Sinkhorn [Scetbon and Cuturi, 2022]. |
Generalized k-means initializer [Scetbon and Cuturi, 2022]. |