Add type annotations for torch.nn.utils.* (#43080)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/43013
Redo of gh-42954
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43080
Reviewed By: albanD
Differential Revision: D23681334
Pulled By: malfet
fbshipit-source-id: 20ec78aa3bfecb7acffc12eb89d3ad833024394c
