ott.solvers.linear.semidiscrete.SemidiscreteOutput.sample

SemidiscreteOutput.sample(rng, num_samples, *, epsilon=None)

Sample a point cloud and compute the OT solution.

Parameters:

• rng (Array) – Random key used for seeding.

• num_samples (int) – Number of samples.

• epsilon (Optional[float]) – Epsilon regularization. If None, use the one stored in the geometry.

Return type:

Union[SinkhornOutput, HardAssignmentOutput]

Returns:

The sampled output.