ott.geometry.pointcloud.PointCloud.marginal_from_scalings#

PointCloud.marginal_from_scalings(u, v, axis=0)#

Output marginal of transportation matrix from scalings.

Return type

Array

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

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

  • axis (int) –