ott.math.velocity_from_brenier_potential

ott.math.velocity_from_brenier_potential(potential, **kwargs)

Get optimal time-dependent velocity field from Brenier potential.

The solution is computed numerically using a Legendre transform.

Parameters:

• potential (Callable[[Array], Array]) – A convex potential of shape [d,].

• kwargs (Any) – Keyword arguments for legendre().

Return type:

Callable[[Array, Array], Array]

Returns:

A time-parameterized velocity field vel(t, x) that expects time array of shape [n,] and inputs of shape [n, d].