randn

cuequivariance_jax.randn(key, rep, leading_shape=(), dtype=None)

Generate a random cuex.RepArray.

Parameters:

• key (jax.Array) – Random key.

• rep (Rep) – representation.

• leading_shape (tuple[int, ...], optional) – Leading shape of the array. Defaults to ().

• dtype (jnp.dtype) – Data type of the array.

Returns:

Random RepArray.

Return type:

RepArray

Example

>>> key = jax.random.key(0)
>>> rep = cue.IrrepsAndLayout(cue.Irreps("O3", "2x1o"), cue.ir_mul)
>>> cuex.randn(key, rep, ())
{0: 2x1o} [...]