ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein.one_iteration

LRGromovWasserstein.one_iteration(ot_prob, state, iteration, compute_error)

Carries out one low-rank GW iteration.

Depending on lse_mode, these iterations can be either in:

• log-space for numerical stability.

• scaling space, using standard kernel-vector multiply operations.

Parameters:

• ot_prob (QuadraticProblem) – the transport problem definition

• state (LRGWState) – the current state.

• iteration (int) – the current iteration of the GW outer loop.

• compute_error (bool) – flag to indicate this iteration computes/stores an error

Return type:

LRGWState

Returns:

The updated state.