3D wavelet transform

[RY-97]

I was doing a similar job recently and encountered the same problem. I would like to ask for your help. I want to input a (b, c, w, h, d) 3D image data. The output after discrete wavelet transform is still consistent in shape, but does not require inverse wavelet transform. Do you have any solution?

[vectorzwt]

You can check this repo:
https://github.com/KeKsBoTer/torch-dwt.