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 whenwarm_start
is False.
- Return type:
GWState
- Returns:
The initial Gromov-Wasserstein state.