nn.SO3ParametricGatedNonlinearity

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

SO3-equivariant parametric gated nonlinearity.

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

\[y_{i,s,f} = x_{j,s,f} * \sigma(f(x_{j,0,\cdot}))\]