Does someone know how to extract outputs at specific locations? For example, I would like to get all outputs corresponding to convolution layers, including the hidden one under sequential. Any idea?
Thanks