|
NEML2 2.1.0
|
Classes | |
| class | ODF |
| class | KDEODF |
| class | Kernel |
| class | DeLaValleePoussinKernel |
Functions | |
| gauss_points (deg, device=torch.device("cpu")) | |
| spherical_quadrature (deg, device=torch.device("cpu")) | |
| rotation_quadrature (deg, device=torch.device("cpu")) | |
| split (X, i) | |
| beta (z1, z2) | |
| beta | ( | z1, | |
| z2 ) |
Calculate the beta function
Args:
z1 (torch.tensor): first input
z2 (torch.tensor): second input
| gauss_points | ( | deg, | |
| device = torch.device("cpu") ) |
Wrap numpy to get Gauss points and weights
Args:
deg (int): degree
Keyword Args:
device (torch.device): which torch device to use
| rotation_quadrature | ( | deg, | |
| device = torch.device("cpu") ) |
Construct a rotational quadrature rule
I am again baking the dV into the weights
Args:
deg (int): degree
Keyword Args:
device (torch.device): which device to run
| spherical_quadrature | ( | deg, | |
| device = torch.device("cpu") ) |
Construct a spherical quadature rule (unit sphere)
Note I'm baking in dV to the weights
Args:
deg (int): degree
Keyword Args:
device (torch.device): which device to use
| split | ( | X, | |
| i ) |
Helper routine to split a batch of orientations into test/validation sets
Args:
X (indexable in first dimension): reference set
i (int): index to separate