ott.tools.conformal.sample_target_measure

ott.tools.conformal.sample_target_measure#

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

Sample target measure for OTCP.

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

  • n_per_radius (Optional[int]) – Optionally specify how many samples by radius

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

Return type:

Tuple[Array, Array]

Returns:

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