title: How does the wavelet transform insert after the flatten layer into the convolution neural network?
body: <p>I'm trying to use Wavelet Transform for data that transfer-learning output of the VGG 16 network. But VGG 16 give 1 dimensional feature vector shape of (4096,). how can i do this?.</p>
  <pre><code>model = keras.Sequential(
          [
              keras.Input(shape=(4096,)),
  
              DWT.DWT('db1', 'smooth'),
  
              layers.Dense(units=4096),
              layers.Dense(units=2048), 
              layers.Dropout(0.5),
              layers.Dense(units=1024),
              layers.Dense(units=2, name='svm'),
          ]
      )
  </code></pre>