Hello,
I have got an autoencoder model, and I want to be able to get the output of each input of every batch in every layer of the module and store it to compare it with something else.
How can I do that
I know that I can use hooks to get the activation of the intermediate layers in my module, now I need to use it for every input!!