ott.solvers.quadratic.gromov_wasserstein.GWOutput.apply#

GWOutput.apply(inputs, axis=0)[source]#

Apply the transport to an array; axis=1 for its transpose.

Return type

Array

Parameters
  • inputs (jax.Array) –

  • axis (int) –