Fix type annotation errors in torch.functional (#43446)
Summary:
Closes gh-42968
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43446
Reviewed By: albanD
Differential Revision: D23280962
Pulled By: malfet
fbshipit-source-id: de5386a95a20ecc814c39cbec3e4252112340b3a
