ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter

class ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter(linear_solver, threshold=0.001, min_iterations=5, max_iterations=50, store_inner_errors=False)

Continuous Wasserstein barycenter solver [Cuturi and Doucet, 2014].

Parameters:

• linear_solver (Union[Sinkhorn, LRSinkhorn])

• threshold (float)

• min_iterations (int)

• max_iterations (int)

• store_inner_errors (bool)

Methods

init_state(bar_prob, bar_size[, x_init, rng])	Initialize the state of the Wasserstein barycenter iterations.
iterations(bar_size, bar_prob, x_init, rng)	Wasserstein barycenter outer loop.
output_from_state(state, bar_prob)	Create an output from a barycenter state.

Attributes

is_low_rank	Whether the solver is low-rank.
rank	Rank of the linear OT solver.