Wavelet CNN implementation using Pytorch

Hello friends. I want to implement a CNN in which the wavelet transformation would take place instead of a convolution operation. Can anyone help me in giving a prototype of a PyTorch code for this deep learning model? This is for the purpose of the experiment of remote sensing image classification.