channelwise_tensor_product#

cuequivariance.group_theory.descriptors.channelwise_tensor_product(
irreps1: Irreps,
irreps2: Irreps,
irreps3_filter=None,
simplify_irreps3: bool = False,
) → EquivariantPolynomial#

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

Construct a channelwise tensor product descriptor.

This operation is computationally sparser than the fully connected tensor product.

Parameters:

  • irreps1 (Irreps) – Irreps of the first operand.

  • irreps2 (Irreps) – Irreps of the second operand.

  • irreps3_filter (sequence of Irrep, optional) – Irreps of the output to consider.

  • simplify_irreps3 (bool, optional) – If True, the irreps of the output are simplified.

Returns:

Descriptor of the channelwise tensor product.

Return type:

cue.EquivariantPolynomial