symmetric_contraction#

cuequivariance.descriptors.symmetric_contraction(
irreps_in: Irreps,
irreps_out: Irreps,
degrees: list[int],
) → EquivariantTensorProduct#

subscripts: weights[u],input[u],output[u]

构建对称收缩的描述符。

对称收缩是输入与其自身进行 degree 次收缩的加权和。

参数:

  • irreps_in (Irreps) – 输入 irreps，多重性并行处理。

  • irreps_out (Irreps) – 输出 irreps。

  • degree (int) – 对称收缩的度数。

返回:

对称收缩的描述符。操作数是权重、输入 degree 次和输出。

返回类型:

cue.EquivariantTensorProduct

示例

>>> cue.descriptors.symmetric_contraction(
...    16 * cue.Irreps("SO3", "0 + 1 + 2"),
...    16 * cue.Irreps("SO3", "0 + 1"),
...    [1, 2, 3]
... )
EquivariantTensorProduct(32x0+80x0+176x0 x (16x0+16x1+16x2)^(1..3) -> 16x0+16x1)

其中 32x0+80x0+176x0 是每个度数所需的权重（度数 1 为 32，度数 2 为 80，度数 3 为 176）。