virne.solver.learning.neural_network.sinkhorn

Classes

SinkhornNetwork([max_iter, tau, epsilon, ...])

SinkhornNetwork algorithm turns the input matrix into a bi-stochastic matrix. Parameter: maximum iterations max_iter a small number for numerical stability epsilon Input: input matrix s Output: bi-stochastic matrix s.