ott.core.continuous_barycenter.WassersteinBarycenter.init_state#

WassersteinBarycenter.init_state(bar_prob, bar_size, x_init=None, rng=0)[source]#

Initialize the state of the Wasserstein barycenter iterations.

Parameters
Return type

BarycenterState

Returns

The initial barycenter state.