[JIT] Update type of the unsqueeze's output in shape analysis. (#40733)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/40733
Test Plan: Imported from OSS
Reviewed By: eellison
Differential Revision: D22298537
Pulled By: ZolotukhinM
fbshipit-source-id: a5d4597ed10bcf14d1b28e914bf898d0cae5b4c0
Author
Mikhail Zolotukhin