Ensure ShardedTensor handles list/tuple appropriately as `size` parameter. (#62109)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62109
The `size` parameter only worked correctly for *args like invocation
:10, 20 and not for list: [10, 20] and tuples: (10, 20). This PR ensures this
works similar to `torch.empty`.
ghstack-source-id: 134246166
Test Plan:
1) unit tests
2) waitforbuildbot
Reviewed By: SciPioneer
Differential Revision: D29884768
fbshipit-source-id: 7a4a3c5ed5d7c081344f6ead3170905b97fc652d