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.
output_from_state(state)

Attributes

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