ott.geometry.graph.Graph.apply_transport_from_potentials#

Graph.apply_transport_from_potentials(f, g, vec, axis=0)[source]#

Since applying from potentials is not feasible in grids, use scalings.

Return type

Array

Parameters

  • f (jax.Array) –

  • g (jax.Array) –

  • vec (jax.Array) –

  • axis (int) –