search

fbcotter / pytorch_wavelets

Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet
Other
943 stars 146 forks source link

Fix bux in lowlevel.py #10

Closed mclaughlin6464 closed 5 years ago

mclaughlin6464 commented 5 years ago

I got a syntax error on line 62 in python 2. I replaced it with a np.expand_dims call, which should do the same operation more generically.

fbcotter commented 5 years ago

Thanks @mclaughlin6464, I've merged but I'd be surprised if this was the only change that needed to be made to make it work with python 2.