nn.SO3Convolution

class nn.SO3Convolution(*args: Any, **kwargs: Any)[source]

SO3-equivariant convolution using Clebsch-Gordon tensor product.

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

\[y_{i,s,f} = \sum_{j,s_1,s_2} x_{j,s_2,f} W_{s_1,f}(r_{ij}) Y_{s_1}(r_{ij}) C_{s_1,s_2}^{s}\]