pytorch

4afe2db6 - [shard] fix sharded_tensor tests

[shard] fix sharded_tensor tests Differential Revision: [D35599369](https://our.internmc.facebook.com/intern/diff/D35599369/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/75712 Approved by: https://github.com/fduwjj