Fix sharded_tensor test_sharded_tensor_to_cpu (#91453)
Fixes https://github.com/pytorch/pytorch/issues/91381
Assert needs to be updated in the test. Run `ciflow/periodic` to run the multigpu tests
Pull Request resolved: https://github.com/pytorch/pytorch/pull/91453
Approved by: https://github.com/clee2000
