[distributed][sharded_tensor] Move local_shards check from ShardedTensorBase to ShardedTensor (#100197)
Differential Revision: [D45369211](https://our.internmc.facebook.com/intern/diff/D45369211)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/100197
Approved by: https://github.com/fduwjj
