from_segments

cuequivariance_jax.from_segments(

irreps: Irreps | str,

segments: Sequence[Array],

shape: tuple[int, ...],

layout: IrrepsLayout | None = None,

dtype: dtype | None = None,

) → RepArray

从分段构造 RepArray。

参数:

• irreps (Irreps) – irreps。

• segments (list of jax.Array) – 分段。

• shape (tuple of int) – 最终数组的形状。

• layout (IrrepsLayout) – 数据布局。

• dtype – 数据类型

返回值:

RepArray。

返回类型:

RepArray

示例

>>> cuex.from_segments(
...     cue.Irreps("SO3", "2x0 + 1"),
...     [jnp.array([[1.0], [2.0]]), jnp.array([[0.0], [0.0], [0.0]])],
...     (-1,), cue.ir_mul)
{0: 2x0+1} [1. 2. 0. 0. 0.]