ott.initializers.nn.initializers.MetaInitializer.init_dual_a#

MetaInitializer.init_dual_a(ot_prob, lse_mode)[source]#

Initialize Sinkhorn potential/scaling f_u.

Parameters
  • ot_prob (LinearProblem) – OT problem between discrete distributions of size n and m.

  • lse_mode (bool) – Return potential if true, scaling if false.

Return type

Array

Returns

potential/scaling, array of size n.