fully_connected_tensor_product#

cuequivariance.descriptors.fully_connected_tensor_product(
irreps1: Irreps,
irreps2: Irreps,
irreps3: Irreps,
) → EquivariantTensorProduct#

subscripts: weights[uvw],lhs[iu],rhs[jv],output[kw]

构建全连接张量积描述符。

该描述符通过迭代输入和输出不可约表示的所有可能组合，并为每个组合构建路径来构建。

参数:

  • irreps1 (Irreps) – 第一个操作数的不可约表示。

  • irreps2 (Irreps) – 第二个操作数的不可约表示。

  • irreps3 (Irreps) – 输出的不可约表示。

返回:

全连接张量积的描述符。

返回类型:

cue.EquivariantTensorProduct

示例

>>> cue.descriptors.fully_connected_tensor_product(
...    16 * cue.Irreps("SO3", "0 + 1 + 2"),
...    16 * cue.Irreps("SO3", "0 + 1 + 2"),
...    16 * cue.Irreps("SO3", "0 + 1 + 2"),
... )
EquivariantTensorProduct(61440x0 x 16x0+16x1+16x2 x 16x0+16x1+16x2 -> 16x0+16x1+16x2)

其中 61440x0 是混合所有输入和所有输出所需的 61440 个权重。