ott.solvers.linear.sinkhorn.SinkhornOutput.apply#

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

Apply the transport to a ndarray; axis=1 for its transpose.

Return type

Array

Parameters
  • inputs (jax.Array) –

  • axis (int) –