ott.core.continuous_barycenter.WassersteinBarycenter#

class ott.core.continuous_barycenter.WassersteinBarycenter(epsilon=None, rank=- 1, linear_ot_solver=None, min_iterations=5, max_iterations=50, threshold=0.001, jit=True, store_inner_errors=False, **kwargs)[source]#

A Continuous Wasserstein barycenter solver, built on generic template.

Methods

init_state(bar_prob, bar_size[, x_init, rng])

Initialize the state of the Wasserstein barycenter iterations.

output_from_state(state)

rtype

BarycenterState

Attributes

is_low_rank

Whether the solver is low-rank.

Parameters