[FSDP][Easy] Save unpadded/padded unsharded sizes as attributes (#84366)
Differential Revision: [D39331199](https://our.internmc.facebook.com/intern/diff/D39331199)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84366
Approved by: https://github.com/rohan-varma