cr.nimble.data.random_subspaces_jit

cr.nimble.data.random_subspaces_jit(key, N, D, K)

Generates a set of orthonormal bases for random low dimensional subspaces

Parameters

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

  • N (int) – Dimension of the ambient space

  • D (int) – Dimension of the low dimensional subspace

  • K (int) – Number of low dimensional subspaces

Returns

A list of orthonormal bases for the random low dimensional subspaces

Return type

(list of jax.numpy.ndarray)