IrrepsLayout

class cuequivariance.IrrepsLayout(

值,

names=<未给定>,

*values,

module=None,

qualname=None,

type=None,

start=1,

boundary=None,

)

可能的数据布局的枚举。

mul_ir

先是重数，然后是不可约表示。此布局对应于库 e3nn 中使用的布局。

ir_mul

先是不可约表示，然后是重数。此布局与 e3nn 中使用的布局不同，但在某些情况下可能更方便。

示例

>>> cue.mul_ir
(mul,irrep)

>>> cue.ir_mul
(irrep,mul)

方法

shape(

mulir: MulIrrep | tuple[int, Irrep],

) → tuple[int, int]

给定布局的张量的形状。

示例

>>> mulir = cue.MulIrrep(32, cue.O3(2, -1))
>>> mulir
32x2o

>>> cue.mul_ir.shape(mulir)
(32, 5)

>>> cue.ir_mul.shape(mulir)
(5, 32)