concatenate

cuequivariance_jax.concatenate(

arrays: list[RepArray],

) → RepArray

Concatenate a list of cuex.RepArray

Parameters:

• arrays (list of RepArray) – List of arrays to concatenate.

• axis (int, optional) – Axis along which to concatenate. Defaults to -1.

Example

>>> with cue.assume(cue.SO3, cue.ir_mul):
...     x = cuex.RepArray("3x0", jnp.array([1.0, 2.0, 3.0]))
...     y = cuex.RepArray("1x1", jnp.array([0.0, 0.0, 0.0]))
>>> cuex.concatenate([x, y])
{0: 3x0+1} [1. 2. 3. 0. 0. 0.]