Fix reshape for the Lazy key (#62846)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62846
Test Plan: CI
Reviewed By: zou3519
Differential Revision: D30162185
Pulled By: asuhan
fbshipit-source-id: d582dcef35ce7e8bebf161a5c93e470339891e29
