ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.init_state#

GromovWasserstein.init_state(prob, init, rng)[source]#

Initialize the state of the Gromov-Wasserstein iterations.

Parameters:

  • prob (QuadraticProblem) – Quadratic OT problem.

  • init (LinearProblem) – Initial linearization of the quadratic problem.

  • rng (PRNGKeyArray) – Random key for low-rank initializers. Only used when warm_start is False.

Return type:

GWState

Returns:

The initial Gromov-Wasserstein state.