Add `torch.movedim` (#41480)
Summary:
https://github.com/pytorch/pytorch/issues/38349 #36048
TODO:
* [x] Tests
* [x] Docs
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41480
Reviewed By: zhangguanheng66
Differential Revision: D22649917
Pulled By: zou3519
fbshipit-source-id: a7f3920a24bae16ecf2ad731698ca65ca3e8c1ce
