ott.tools.progot.ProgOTOutput.transport

ott.tools.progot.ProgOTOutput.transport#

ProgOTOutput.transport(x, num_steps=None, return_intermediate=False)[source]#

Transport points.

Parameters:
  • x (Array) – Array of shape [n, d] to transport.

  • num_steps (Optional[int]) – Number of steps. If None, use the full number of steps.

  • return_intermediate (bool) – Whether to return intermediate values.

Return type:

Tuple[Array, Array]

Returns:

  • If return_intermediate = True, return arrays of shape [num_steps, n, d] and [num_steps, n, d] corresponding to the interpolations and push-forwards after each step, respectively.

  • Otherwise, return arrays of shape [n, d] and [n, d] corresponding to the last interpolation and push-forward, respectively.