Wavelet-Packet Dimension patch

v0lta / PyTorch-Wavelet-Toolbox

Differentiable fast wavelet transforms in PyTorch with GPU support.

https://pytorch-wavelet-toolbox.readthedocs.io

European Union Public License 1.2

279 stars 36 forks source link

Closed v0lta closed 2 years ago

v0lta commented 2 years ago

I promised not to commit to master anymore. So here is the pull request to fix the dimension problem from issue https://github.com/v0lta/PyTorch-Wavelet-Toolbox/issues/21 . I have:

Gone over our usage of tf.squeeze and
Simplified and extended test coverage for batched packet transforms, including batch_size = 1. Which used to clash with our bare tf.squeeze statements.

felixblanke commented 2 years ago

I would propose some changes. See #24