ott.initializers.linear.initializers_lr.RandomInitializer

ott.initializers.linear.initializers_lr.RandomInitializer#

class ott.initializers.linear.initializers_lr.RandomInitializer(rank, **kwargs)[source]#

Low-rank Sinkhorn factorization using random factors.

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.