elementwise_tensor_product#

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

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

Construct an elementwise tensor product descriptor.

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.

Returns:

Descriptor of the elementwise tensor product.

Return type:

cue.EquivariantPolynomial