ott.experimental.mmsinkhorn.MMSinkhorn.init_state

ott.experimental.mmsinkhorn.MMSinkhorn.init_state#

MMSinkhorn.init_state(n_s)[source]#

Return the initial state of the loop.

Parameters:

n_s (Tuple[int, ...])

Return type:

MMSinkhornState