ott.geometry.grid.Grid.apply_transport_from_potentials#

Grid.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._src.basearray.Array) –

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

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

  • axis (int) –