encode_rotation_angle

cuequivariance_torch.encode_rotation_angle(angle: Tensor, ell: int) → Tensor

将角度编码为余弦和正弦张量。

编码方式为

[cos(l * angle), cos((l - 1) * angle), ..., cos(angle), 1, sin(angle), sin(2 * angle), ..., sin(l * angle)].

此编码用于馈送执行旋转的分段张量积。