Closed tanvir9476 closed 2 years ago
What's the shape of your input tensors (block_input and x). The library expects a 4-dimensional tensor with the last three components defined (e.g., [None, 128, 128, 3] for RGB 128x128 pixel images)
P.S. Note that this is not a discrete Fourier transform (DFT) library, but a Wavelet one.
I think it's (None, None, None, 64)
; that's why it was throwing an error. Didn't know the library expected tensors with last 3 components explicitly defined. Thanks for the clarification.
I've been trying to use the DWT and IDWT layer inside a residual module:
The model was used on an image dataset.