spherical_harmonics

cuequivariance_jax.spherical_harmonics(

ls: list[int],

vector: RepArray,

normalize: bool = True,

) → RepArray

Compute the spherical harmonics of a vector.

Parameters:

• ls (list of int) – List of spherical harmonic degrees.

• vector (RepArray) – Input vector(s).

• normalize (bool) – Whether to normalize the vector before computing the spherical harmonics.

• algorithm (str) – Algorithm to use for the tensor product. See cuex.tensor_product for more information.

Returns:

Spherical harmonics of the vector.

Return type:

RepArray