[vmap] implement batching rule for fill_ and zero_ (#48516)
Summary:
Fix https://github.com/pytorch/pytorch/issues/47755
- This PR implements batching rules for in-place operators `fill_` and `zero_`.
- Testcases are added to the `test/test_vmap.py`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/48516
Reviewed By: H-Huang
Differential Revision: D25431557
Pulled By: zou3519
fbshipit-source-id: 437b0534dc0b818fbe05f7fcfcb649aa677483dc