tfg.math.spherical_harmonics.rotate_zonal_harmonics

View source on GitHub

Rotates zonal harmonics.

Note:

In the following, A1 to An are optional batch dimensions.

zonal_coeffs A tensor of shape [C] storing zonal harmonics coefficients.
theta A tensor of shape [A1, ..., An, 1] storing polar angles.
phi A tensor of shape [A1, ..., An, 1] storing azimuthal angles.
name A name for this op. Defaults to 'spherical_harmonics_rotate_zonal_harmonics'.

A tensor of shape [A1, ..., An, C*C] storing coefficients of the rotated harmonics.

ValueError If the shape of zonal_coeffs, theta or phi is not supported.