nn.SO3TensorProduct

class nn.SO3TensorProduct(*args: Any, **kwargs: Any)

SO3-equivariant Clebsch-Gordon tensor product.

With combined indexing s=(l,m), this can be written as:

\[y_{s,f} = \sum_{s_1,s_2} x_{2,s_2,f} x_{1,s_2,f} C_{s_1,s_2}^{s}\]