ott.core.sinkhorn.Sinkhorn.init_state

Sinkhorn.init_state(ot_prob, init)

Return the initial state of the loop.

Return type

SinkhornState

Parameters

• ot_prob (ott.core.linear_problems.LinearProblem) –

• init (Tuple[jax._src.numpy.ndarray.ndarray, jax._src.numpy.ndarray.ndarray]) –