ott.initializers.linear.initializers_lr.Rank2Initializer.from_solver

ott.initializers.linear.initializers_lr.Rank2Initializer.from_solver#

classmethod Rank2Initializer.from_solver(solver, *, kind, **kwargs)#

Create a low-rank initializer from a linear or quadratic solver.

Parameters:
  • solver (Union[LRSinkhorn, LRGromovWasserstein]) – Low-rank linear or quadratic solver.

  • kind (Literal['random', 'rank2', 'k-means', 'generalized-k-means']) – Which initializer to instantiate.

  • kwargs (Any) – Keyword arguments when creating the initializer.

Return type:

LRInitializer

Returns:

Low-rank initializer.