Allow resizing of parametrized tensors (#60418)
Summary:
Modify `parametrize.py` to allow resizing of parametrized tensors
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60418
Test Plan:
`buck test mode/dev-nosan //caffe2/test:nn -- --exact 'caffe2/test:nn - test_register_and_remove_parametrization (test_nn.TestNN)'`
https://pxl.cl/1L0wh
Reviewed By: z-a-f
Differential Revision: D29279442
Pulled By: kazhou
fbshipit-source-id: 4d94915748f896e7761a40ad18f4c6444f505c3a