randn

cuequivariance_jax.randn(

key: Array,

rep: Rep,

leading_shape: tuple[int, ...] = (),

dtype: dtype | None = None,

) → RepArray

生成一个随机的 cuex.RepArray。

参数:

• key (jax.Array) – 随机键。

• rep (Rep) – 表示。

• leading_shape (tuple[int, ...], 可选) – 数组的前导形状。默认为 ()。

• dtype (jnp.dtype) – 数组的数据类型。

返回:

随机 RepArray。

返回类型:

RepArray

示例

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