Would there be a way to use batches?

Hello, and thanks again for your code!

I would like to use your tfw functions to compute the wavelet transform of a whole batch of images in one shot. I have tried but I don't see a way to do so given that your dwt2d relies on dwt1d, which already uses the rows of the 2d image as batches (if I understood correctly). Would you have a suggestion to help me? 

Thank you!