ott.initializers.linear.initializers.SubsampleInitializer.init_fu

ott.initializers.linear.initializers.SubsampleInitializer.init_fu#

SubsampleInitializer.init_fu(ot_prob, lse_mode, rng=None)[source]#

Initialize Sinkhorn potential/scaling f_u.

Parameters:

  • ot_prob (LinearProblem) – Linear OT problem.

  • lse_mode (bool) – Return potential if True, scaling if False.

  • rng (Optional[Array]) – Random number generator for stochastic initializers.

Return type:

Array

Returns:

potential/scaling, array of size [n,].