ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter#
- class ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter(epsilon=None, rank=-1, linear_ot_solver=None, min_iterations=5, max_iterations=50, threshold=0.001, store_inner_errors=False, **kwargs)[source]#
Continuous Wasserstein barycenter solver [Cuturi and Doucet, 2014].
- Parameters
Methods
init_state
(bar_prob, bar_size[, x_init, rng])Initialize the state of the Wasserstein barycenter iterations.
output_from_state
(state)- param state
Attributes
Whether the solver is low-rank.