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 when warm_start is False.

Return type

GWState

Returns

The initial Gromov-Wasserstein state.