as_irreps_array

cuequivariance_jax.as_irreps_array(input, layout=None, like=None)

Converts input to a RepArray. Arrays are assumed to be scalars.

Examples

>>> with cue.assume(cue.O3):
...     cuex.as_irreps_array([1.0], layout=cue.ir_mul)
{0: 0e} [1.]

Parameters:

• input (Any)

• layout (IrrepsLayout | None)

• like (RepArray | None)

Return type:

RepArray