elementwise_tensor_product

cuequivariance.descriptors.elementwise_tensor_product(

irreps1: Irreps,

irreps2: Irreps,

irreps3_filter: Sequence[Irrep] | None = None,

) → EquivariantTensorProduct

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.EquivariantTensorProduct