You can use register_forward_hook for whichever intermediate layer you want. Check out my previous post below on extracting features.
register_forward_hook