ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein.one_iteration#
- LRGromovWasserstein.one_iteration(ot_prob, state, iteration, compute_error)[source]#
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 definitionstate (
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.