ott.tools.progot.ProgOTOutput.get_output

ott.tools.progot.ProgOTOutput.get_output#

ProgOTOutput.get_output(step)[source]#

Get the OT solver output at a given step.

Parameters:

step (int) – Iteration step in \([0, \text{num_steps})\).

Return type:

Union[SinkhornOutput, SinkhornDivergenceOutput]

Returns:

The OT solver output at a step.