ott.solvers.linear.sinkhorn.Sinkhorn.init_state

Sinkhorn.init_state(ot_prob, init)

Return the initial state of the loop.

Return type

SinkhornState

Parameters

• ot_prob (ott.problems.linear.linear_problem.LinearProblem) –

• init (Tuple[jax.Array, jax.Array]) –