ott.core.initializers.SortingInitializer#

class ott.core.initializers.SortingInitializer(vectorized_update=True, tolerance=0.01, max_iter=100)[source]#

Sorting initializer [Thornton and Cuturi, 2022].

Solves non-regularized OT problem via sorting, then compute potential through iterated minimum on C-transform and use this potential to initialize regularized potential.

Parameters
  • vectorized_update (bool) – Use vectorized inner loop if true.

  • tolerance (float) – DualSort convergence threshold.

  • max_iter (int) – Max DualSort steps.

Methods

init_dual_a(ot_prob, lse_mode[, init_f])

Apply DualSort algorithm.

init_dual_b(ot_prob, lse_mode)

Initialize Sinkhorn potential/scaling g_v.