Torch Trainable Parameters Vs Statistics