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