ott.core.gromov_wasserstein.GromovWasserstein#

class ott.core.gromov_wasserstein.GromovWasserstein(*args, warm_start=None, quad_initializer=None, kwargs_init=None, **kwargs)[source]#

Gromov-Wasserstein solver.

Parameters

Methods

create_initializer(prob)

Create quadratic, possibly low-rank initializer.

init_state(prob, init, key)

Initialize the state of the Gromov-Wasserstein iterations.

output_from_state(state)

Create an output from a loop state.

Attributes

is_low_rank

Whether the solver is low-rank.

warm_start

Whether to initialize (low-rank) Sinkhorn using previous solutions.