spherical_harmonics#

cuequivariance_jax.spherical_harmonics( ls: list[int], vector: RepArray, normalize: bool = True, algorithm: str = 'stacked', ) → RepArray#

计算向量的球谐函数。

参数:

ls (list of int) – 球谐函数阶数的列表。
vector (RepArray) – 输入向量。
normalize (bool) – 是否在计算球谐函数之前对向量进行归一化。
algorithm (str) – 用于张量积的算法。更多信息请参见 cuex.tensor_product。

返回:

向量的球谐函数。

返回类型: