ott.solvers.linear.sinkhorn.SinkhornState.recenter#
- SinkhornState.recenter(f, g, ot_prob)[source]#
Re-center dual potentials.
If the
ot_prob
is balanced, thef
potential is zero-centered. Otherwise, use prop. 2 of [Sejourne et al., 2022] re-center the potentials ifftau_a < 1
andtau_b < 1
.