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)