ott.tools.transport.Transport.apply#

Transport.apply(inputs, axis=0)[source]#
Return type

Array

Parameters
  • inputs (jax._src.basearray.Array) –

  • axis (int) –