from_segments#

cuequivariance_jax.from_segments(irreps, segments, shape, layout=None, dtype=None)#

Construct a RepArray from segments.

Parameters:
Returns:

the RepArray.

Return type:

RepArray

Examples

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