heat
heat copied to clipboard
Implement batch Wiener filter
Feature functionality
scipy.signal.wiener applies a Wiener denoising filter to all dimensions of an n-D array. We want to be able to apply a n-D Wiener filter to a batch of n-D arrays, i.e. to a (n+1)-D array with axis 0 being the batch dimension (and potentially the split axis of a distributed DNDarray).
scipy wiener source code is here.
Vectorized scipy.signal.correlate
can be performed by torch.conv1d
& co.
Related RFI mitigation use case