encode_rotation_angle

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

Encode a angle into a tensor of cosines and sines.

The encoding is:

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

This encoding is used to feed the segmented tensor products that perform rotations.