ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer.init_q#
- GeneralizedKMeansInitializer.init_q(ot_prob, rng, *, init_g, **kwargs)#
Initialize the low-rank factor \(Q\).
- Parameters:
ot_prob (
Union[LinearProblem,QuadraticProblem]) – OT problem.rng (
PRNGKeyArray) – Random key for seeding.init_g (
Array) – Initial value for \(g\) factor.kwargs (
Any) – Additional keyword arguments.
- Return type:
- Returns:
Array of shape
[n, rank].