ott.initializers.linear.initializers.SinkhornInitializer#

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

Base class for Sinkhorn initializers.

Methods

init_dual_a(ot_prob, lse_mode)

Initialization for Sinkhorn potential/scaling f_u.

init_dual_b(ot_prob, lse_mode)

Initialization for Sinkhorn potential/scaling g_v.