Thank you for your response.
extracted_feat[str(key)]: torch.Size([1, 64, 600, 1200])
extracted_feat_aug[str(key)]: torch.Size([1, 64, 600, 1200])
what I am trying to do is subtract the activations of relu layers for clean image and augmented image then  divide it by the mean value of clean activations of the same layer, then compute the L2 norm of it.