ott.initializers.linear.initializers_lr.Rank2Initializer

class ott.initializers.linear.initializers_lr.Rank2Initializer(rank, **kwargs)

Low-rank Sinkhorn factorization using rank-2 factors [Scetbon et al., 2021].

Parameters:

• rank (int) – Rank of the factorization.

• kwargs (Any) – Additional keyword arguments.

Methods

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	Rank of the transport matrix factorization.