I have successfully trained a VAE on image data so that it is able to reconstruct the images. I want to use the Encoder of the VAE as a feature extractor in my framework. Th