as_irreps_array

cuequivariance_jax.as_irreps_array(

input: Any,

layout: IrrepsLayout | None = None,

like: RepArray | None = None,

) → RepArray

将输入转换为 RepArray。 数组被假定为标量。

示例

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