cuequivariance.NumpyIrrepsArray

class cuequivariance.NumpyIrrepsArray(irreps: Irreps | str, array: ndarray, layout: IrrepsLayout | None = None)

Data Array transforming according to the irreducible representations of a Lie group.

Parameters:

• irreps – cuequivariance.Irreps or str Irreps of the data array.

• array – numpy.ndarray Data array.

• layout – cuequivariance.IrrepsLayout, optional Layout of the data array.

__init__(irreps: Irreps | str, array: ndarray, layout: IrrepsLayout | None = None)

Methods

__init__(irreps, array[, layout])
change_layout(layout)
filter(*[, keep, drop])
merge_consecutive()
regroup()
reshape(shape)
simplify()
sort()

Attributes

dtype	Data type of the data array.
ndim	Number of dimensions of the data array.
segments	List of segments of the data array.
shape	Shape of the data array.
irreps
array
layout