ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer.init_r#

GeneralizedKMeansInitializer.init_r(ot_prob, rng, *, init_g, **kwargs)#

Initialize the low-rank factor \(R\).

Parameters
  • ot_prob (Union[LinearProblem, QuadraticProblem]) – Linear OT problem.

  • rng (PRNGKeyArray) – Random key for seeding.

  • init_g (Array) – Initial value for \(g\) factor.

  • kwargs (Any) – Additional keyword arguments.

Return type

Array

Returns

Array of shape [m, rank].