core.models.scn.spherical_harmonics

core.models.scn.spherical_harmonics#

Copyright (c) Meta, Inc. and its affiliates.

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

Attributes#

_Jd

Classes#

SphericalHarmonicsHelper

Helper functions for spherical harmonics calculations and representations

Functions#

wigner_D(l, alpha, beta, gamma)

_z_rot_mat(angle, l)

Module Contents#

core.models.scn.spherical_harmonics._Jd#
class core.models.scn.spherical_harmonics.SphericalHarmonicsHelper(lmax: int, mmax: int, num_taps: int, num_bands: int)#

Helper functions for spherical harmonics calculations and representations

Parameters:

  • lmax (int) – Maximum degree of the spherical harmonics

  • mmax (int) – Maximum order of the spherical harmonics

  • num_taps (int) – Number of taps or rotations (1 or otherwise set automatically based on mmax)

  • num_bands (int) – Number of bands used during message aggregation for the 1x1 pointwise convolution (1 or 2)

lmax#
mmax#
num_taps#
num_bands#
sphere_basis#
sphere_basis_reduce#
InitWignerDMatrix(edge_rot_mat) → None#
InitYRotMapping()#
ToGrid(x, channels) → torch.Tensor#
FromGrid(x_grid, channels) → torch.Tensor#
CombineYRotations(x) → torch.Tensor#
Rotate(x) → torch.Tensor#
FlipGrid(grid, num_channels: int) → torch.Tensor#
RotateInv(x) → torch.Tensor#
RotateWigner(x, wigner) → torch.Tensor#
RotationMatrix(rot_x: float, rot_y: float, rot_z: float) → torch.Tensor#
RotationToWignerDMatrix(edge_rot_mat, start_lmax, end_lmax)#
core.models.scn.spherical_harmonics.wigner_D(l, alpha, beta, gamma)#
core.models.scn.spherical_harmonics._z_rot_mat(angle, l)#
Contents