ott.initializers.linear.initializers_lr.KMeansInitializer.init_r

ott.initializers.linear.initializers_lr.KMeansInitializer.init_r#

KMeansInitializer.init_r(ot_prob, rng, *, init_g, **kwargs)[source]#

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

Parameters:
Return type:

Array

Returns:

Array of shape [m, rank].