pytorch
81528d4b - [shard] add more tensor creation ops (#77185)

Commit
2 years ago
[shard] add more tensor creation ops (#77185) Summary: This PR fix some existing tensor constructors (i.e. full), add sharded_tensor.randn, and add more tensor-like creation ops (i.e. full_like, rand_like, etc.) Test Plan: test_create_sharded_tensor_with_rand test_create_sharded_tensor_like Differential Revision: D36274148 Pull Request resolved: https://github.com/pytorch/pytorch/pull/77185 Approved by: https://github.com/pritamdamania87, https://github.com/fduwjj
Author
Committer
Parents
Loading