In 0.4.1, the jit argument in solvers will be removed. Please jit the solvers explicitly.

ott.solvers.linear.sinkhorn.Sinkhorn.init_state

Contents

ott.solvers.linear.sinkhorn.Sinkhorn.init_state#

Sinkhorn.init_state(ot_prob, init)[source]#

Return the initial state of the loop.

Parameters:

ot_prob (LinearProblem) –
init (Tuple[Array, Array]) –

Return type:

SinkhornState