Fix type annotations for a number of torch.utils submodules (#42711)
Summary:
Related issue on `torch.utils` type annotation hiccups: gh-41794
Pull Request resolved: https://github.com/pytorch/pytorch/pull/42711
Reviewed By: mrshenli
Differential Revision: D23005434
Pulled By: malfet
fbshipit-source-id: 151554b1e7582743f032476aeccdfdad7a252095
