Currently working on a classifier using PyWavelets, here is my calculation block:class WaveletLayer(nn