ott.problems.linear.semidiscrete_linear_problem.SemidiscreteLinearProblem.sample

ott.problems.linear.semidiscrete_linear_problem.SemidiscreteLinearProblem.sample#

SemidiscreteLinearProblem.sample(rng, num_samples, *, epsilon=None)[source]#

Sample a linear OT problem.

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

  • num_samples (int) – Number of samples.

  • epsilon (Optional[float]) – Epsilon regularization. If None, use epsilon.

Return type:

LinearProblem

Returns:

The sampled linear problem.