from_segments#
- cuequivariance_jax.from_segments(irreps, segments, shape, layout=None, dtype=None)#
Construct a RepArray from segments.
- Parameters:
- Returns:
the RepArray.
- Return type:
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.]