ott.initializers.linear.initializers_lr.Rank2Initializer.init_q

ott.initializers.linear.initializers_lr.Rank2Initializer.init_q#

Rank2Initializer.init_q(ot_prob, rng, *, init_g, **kwargs)[source]#

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

Parameters:
Return type:

Array

Returns:

Array of shape [n, rank].