数据布局#

当表示具有重数的不可约表示集合时,有两种方法可以在内存中组织数据

  • (ir, mul) - 不可约表示是最外层维度。

  • (mul, ir) - 重数是最外层维度。这是 e3nn 使用的布局。

Illustration of data layouts

在上面的示例中,所有块的重数均为 4。给定不可约表示的维度,它可能对应于不可约表示 “4x0e + 4x1e + 4x2e”。