I have a neural network model in pytorc which is trained on a dataset. Now, I would like to remove / delete zero parameters weights from Dense layers and Convolution layers. Is there a way to do that in pytorch ? Please help. Thank you.