torch_harmonics.filter_basis.PiecewiseLinearFilterBasis#

class torch_harmonics.filter_basis.PiecewiseLinearFilterBasis(kernel_shape)[source]#

Bases: FilterBasis

Tensor-product basis on a disk constructed from piecewise linear basis functions.

Parameters:

kernel_shape (int | Tuple[int] | Tuple[int, int])

property kernel_size#

Compute the number of basis functions in the kernel.

compute_support_vals(r, phi, r_cutoff)[source]#

Computes the index set that falls into the kernel’s support and returns both indices and values.

Parameters: