ott.solvers.linear.sinkhorn.Sinkhorn.init_state
ott.solvers.linear.sinkhorn.Sinkhorn.init_state#
- Sinkhorn.init_state(ot_prob, init)[source]#
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]) –