ott.initializers.linear.initializers.SubsampleInitializer.init_dual_a#
- SubsampleInitializer.init_dual_a(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 ifTrue
, scaling ifFalse
.rng (
Optional
[PRNGKeyArray
]) – Random number generator for stochastic initializers.
- Return type
- Returns
potential/scaling, array of size
[n,]
.