e3nn_mlx.IrrepsArray

class e3nn_mlx.IrrepsArray(irreps: 'Irreps | str', array: 'Any')[source]
Parameters:
__init__(irreps, array)[source]
Parameters:

  • irreps (Irreps | str)

  • array (Any)

Return type:

None

Methods

__init__(irreps, array)

chunk_arrays()

chunk_dict()

from_chunks(irreps, chunks, *, backend)

regroup()

simplify()

with_array(array)

Attributes

irreps

array

leading_shape

chunks

dtype

shape