ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.init_state
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.init_state#
- GromovWasserstein.init_state(prob, init, key)[source]#
Initialize the state of the Gromov-Wasserstein iterations.
- Parameters
prob (
QuadraticProblem
) – Quadratic OT problem.init (
LinearProblem
) – Initial linearization of the quadratic problem.key (
Array
) – Random key for low-rank initializers. Only used whenwarm_start
is False.
- Return type
GWState
- Returns
The initial Gromov-Wasserstein state.