Fetch values of the latent space

[usarawgi911]

Hi @arashsaber
Thank you for the comprehensive explanation of the code. Can you let me know how do i fetch values of the compressed representation (latent space), which is 'fc2 layer' as i understand? I have been trying out code snippets for the same, but couldn't get the values yet.
Thank you

[arashsaber]

one quick solution is for the function to also output 'fc2'. A better solution is like the one I used for VAE and AAE available in the 'nnexplore' repository. That is, you define the encoder and decoder functions separately and fetch the output of encoder for any input you desire.