Add type annotation logic for complex numbers (#50884)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50884
Test Plan: Imported from OSS
Reviewed By: heitorschueroff
Differential Revision: D26086963
fbshipit-source-id: f103f7f529d63d701c4f17862e30eafbab7d0c68
