escn_tp

cuequivariance.descriptors.escn_tp(

irreps_in: Irreps,

irreps_out: Irreps,

m_max: int | None = None,

l_max: int | None = None,

) → EquivariantTensorProduct

subsrcipts: weights[uv],input[u],output[v]

Tensor Product part of the eSCN convolution introduced in https://arxiv.org/pdf/2302.03655.pdf

Parameters:

• irreps_in (Irreps) – Irreps of the input.

• irreps_out (Irreps) – Irreps of the output.

• m_max (int, optional) – Maximum angular resolution around the principal axis.

• l_max (int, optional) – Maximum angular resolution along the principal axis.

Returns:

Descriptor of the tensor product part of the eSCN convolution.

• Operand 0: weights

• Operand 1: input

• Operand 2: output

Return type:

EquivariantTensorProduct