逐元素张量积#

cuequivariance.descriptors.elementwise_tensor_product(
irreps1: Irreps,
irreps2: Irreps,
irreps3_filter: Sequence[Irrep] | None = None,
) EquivariantTensorProduct#

subscripts: lhs[iu],rhs[ju],output[ku]

构建一个逐元素张量积描述符。

参数:
  • irreps1 (Irreps) – 第一个操作数的 Irreps。

  • irreps2 (Irreps) – 第二个操作数的 Irreps。

  • irreps3_filter (Irrep 序列, 可选) – 要考虑的输出的 Irreps。

返回:

逐元素张量积的描述符。

返回类型:

cue.EquivariantTensorProduct