Add type annotations for a few torch.nn.modules (#46013)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/46012
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46013
Reviewed By: gchanan
Differential Revision: D25012419
Pulled By: ezyang
fbshipit-source-id: 9fd8ad9fa3122efa294a08010171cb7ddf752778
