cr.sparse.data.uniform_points_on_subspaces

cr.sparse.data.uniform_points_on_subspaces(key, bases, n)[source]

Generates a set of nk points on the unit sphere of each of the subspaces

Parameters
  • key – a PRNG key used as the random key.

  • bases (list of jax.numpy.ndarray) – List of orthonormal bases for the low dimensional subspaces

  • n (int) – Number of points on each subspace unit sphere

Returns

A matrix containing the list of points

Return type

(jax.numpy.ndarray)