ott.initializers.linear.initializers_lr.Rank2Initializer#
- class ott.initializers.linear.initializers_lr.Rank2Initializer(rank, **kwargs)[source]#
Low-rank Sinkhorn factorization using rank-2 factors [Scetbon et al., 2021].
Methods
from_solver
(solver, *, kind, **kwargs)Create a low-rank initializer from a linear or quadratic solver.
init_g
(ot_prob, rng, **kwargs)Initialize the low-rank factor \(g\).
init_q
(ot_prob, rng, *, init_g, **kwargs)Initialize the low-rank factor \(Q\).
init_r
(ot_prob, rng, *, init_g, **kwargs)Initialize the low-rank factor \(R\).
Attributes
Rank of the transport matrix factorization.