nn.RealSphericalHarmonics

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

Generates the real spherical harmonics for a batch of vectors.

Note

The vectors passed to this layer are assumed to be normalized to unit length.

Spherical harmonics are generated up to angular momentum lmax in dimension 1, according to the following order: - l=0, m=0 - l=1, m=-1 - l=1, m=0 - l=1, m=1 - l=2, m=-2 - l=2, m=-1 - etc.

Parameters:

lmax – maximum angular momentum
dtype_str – dtype for spherical harmonics coefficients