concatenate

cuequivariance_jax.concatenate(

arrays: list[RepArray],

) → RepArray

连接 cuex.RepArray 列表

参数:

• arrays (list of RepArray) – 要连接的数组列表。

• axis (int, optional) – 沿其连接的轴。默认为 -1。

示例

>>> 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.]