Apologies my bad, pasted wrong thing 
Ment to paste your way
net.module.features[2].register_forward_hook(save_activation('features.2'))

like that I am accessing tensor of this particular activated layer right? So I can compute stats.