[TorchTidy] Adding testing for size and dtype capture (#80071)
Differential Revision: [D37571571](https://our.internmc.facebook.com/intern/diff/D37571571)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80071
Approved by: https://github.com/robieta