ott.initializers.linear.initializers.DefaultInitializer

ott.initializers.linear.initializers.DefaultInitializer#

class ott.initializers.linear.initializers.DefaultInitializer[source]#

Default initialization of Sinkhorn dual potentials/primal scalings.

Methods

init_dual_a(ot_prob, lse_mode[, rng])

Initialize Sinkhorn potential/scaling f_u.

init_dual_b(ot_prob, lse_mode[, rng])

Initialize Sinkhorn potential/scaling g_v.