ott.initializers.linear#
Initializers for linear OT problems, focusing on Sinkhorn and low-rank solvers.
Sinkhorn Initializers#
Default initialization of Sinkhorn dual potentials/primal scalings. |
|
Gaussian initializer [Thornton and Cuturi, 2022]. |
|
Sorting initializer [Thornton and Cuturi, 2022]. |
|
|
Subsample initializer [Thornton and Cuturi, 2022]. |
Low-rank Sinkhorn Initializers#
|
Base class for low-rank 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]. |