ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.apply#

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

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

Return type

Array

Parameters

  • inputs (jax.Array) –

  • axis (int) –