- PointCloud.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 stabilise computations, and lifts this with an exp to recover either of the marginals corresponding to the transport map induced by potentials.
Array) – jnp.ndarray [num_a,] , potential of size num_rows of cost_matrix
Array) – jnp.ndarray [num_b,] , potential of size num_cols of cost_matrix
int) – axis along which to integrate, returns marginal on other axis.
- Return type
a vector of marginals of the transport matrix.