Peter,
I  extracted the Maxpool2d layer (36) and I m aware of the fact that if I give a 3x448x448 image I will get back the 512x14x14 tensor.
I want to use this tensor given by the Maxpool2d layer (36).
Please help.
Do I need to use register hook. If so how?