Named inference rules for some initializer fns
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/22972
Test Plan:
- [namedtensor ci]
Imported from OSS
Differential Revision: D16342782
Pulled By: zou3519
fbshipit-source-id: 25277688ab51e1e98af0e19aeb9c79399171d2fb