ott.geometry.grid.Grid.marginal_from_potentials#
- Grid.marginal_from_potentials(f, g, axis=0)#
Output marginal of transportation matrix from potentials.
This applies first lse kernel in the standard way, removes the correction used to stabilize computations, and lifts this with an exp to recover either of the marginals corresponding to the transport map induced by potentials.
- Parameters
- Return type
- Returns
a vector of marginals of the transport matrix.