channelwise_tensor_product#

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

subscripts: weights[uv],lhs[iu],rhs[jv],output[kuv]

构建通道式张量积描述符。

此操作的计算量比全连接张量积更稀疏。

参数:

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

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

  • irreps3_filter (sequence of Irrep, optional) – 要考虑的输出的不可约表示。

返回:

通道式张量积的描述符。

返回类型:

cue.EquivariantTensorProduct