ott.solvers.linear.continuous_barycenter.WassersteinBarycenter

class ott.solvers.linear.continuous_barycenter.WassersteinBarycenter(epsilon=None, rank=- 1, linear_ot_solver=None, min_iterations=5, max_iterations=50, threshold=0.001, store_inner_errors=False, **kwargs)

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

• epsilon (Optional[float]) –

• rank (int) –

• linear_ot_solver (Optional[Union[sinkhorn.Sinkhorn, sinkhorn_lr.LRSinkhorn]]) –

• min_iterations (int) –

• max_iterations (int) –

• threshold (float) –

• store_inner_errors (bool) –

• kwargs (Any) –