ott.tools.conformal.sobol_ball_sampler

ott.tools.conformal.sobol_ball_sampler#

ott.tools.conformal.sobol_ball_sampler(rng, shape, n_per_radius=None)[source]#

Sample target measure for OTCP.

Parameters:
  • rng (Optional[Array]) – Random number generator.

  • shape (Tuple[int, int]) – Tuple of [n_samples, dim].

  • n_per_radius (Optional[int]) – Optionally specify how many samples to sample per radius.

Return type:

Tuple[Array, Array]

Returns:

Points of shape [n_S * n_R, dim] and weights [n_S * n_R,].