cr.nimble.dsp.randomize_rows

cr.nimble.dsp.randomize_rows(key, X)[source]

Randomizes the rows in X

Parameters

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

  • X (jax.numpy.ndarray) – A 2D data matrix

Returns

The data matrix with randomized rows

Return type

(jax.numpy.ndarray)