ott.geometry.geometry.Geometry.marginal_from_scalings#

Geometry.marginal_from_scalings(u, v, axis=0)[source]#

Output marginal of transportation matrix from scalings.

Return type

Array

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

  • v (jax._src.basearray.Array) –

  • axis (int) –