cr.nimble.data.uniform_points_on_subspaces_jit

cr.nimble.data.uniform_points_on_subspaces_jit(key, bases, n)

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)