Add some missing types to torch (#55184)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55184
Test Plan: Sandcastle
Reviewed By: ezyang
Differential Revision: D27515470
fbshipit-source-id: 264bc067db8fb430465d14bf9508ac8b1faf0f2f
