Commit 04e53f78 authored by Boris Bonev's avatar Boris Bonev
Browse files

adding one-liner in filter basis

parent 013f9a19
...@@ -195,6 +195,7 @@ class PiecewiseLinearFilterBasis(FilterBasis): ...@@ -195,6 +195,7 @@ class PiecewiseLinearFilterBasis(FilterBasis):
return iidx, vals return iidx, vals
def compute_support_vals(self, r: torch.Tensor, phi: torch.Tensor, r_cutoff: float): def compute_support_vals(self, r: torch.Tensor, phi: torch.Tensor, r_cutoff: float):
"""Computes the index set that falls into the kernel's support and returns both indices and values."""
if self.kernel_shape[1] > 1: if self.kernel_shape[1] > 1:
return self._compute_support_vals_anisotropic(r, phi, r_cutoff=r_cutoff) return self._compute_support_vals_anisotropic(r, phi, r_cutoff=r_cutoff)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment