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]

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.

Returns:

Descriptor of the channelwise tensor product.

Return type:

cue.EquivariantTensorProduct