ott.tools.sliced.random_proj_sphere

ott.tools.sliced.random_proj_sphere#

ott.tools.sliced.random_proj_sphere(x, n_proj=1000, rng=None)[source]#

Project data on directions sampled randomly from sphere.

Parameters:
  • x (Array) – Array of size [n, dim].

  • n_proj (int) – Number of randomly generated projections.

  • rng (Optional[Array]) – Key used to sample feature extractors.

Return type:

Array

Returns:

Array of size [n, n_proj] features.