ott.initializers.linear.initializers_lr.Rank2Initializer

Contents

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].

Parameters:
  • rank (int) – Rank of the factorization.

  • kwargs (Any) – Additional keyword arguments.

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

Rank of the transport matrix factorization.