Extend support for exporting reshape to onnx. (#16971)
Summary:
Resolve issue with reshape_as test case.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16971
Differential Revision: D14098871
Pulled By: houseroad
fbshipit-source-id: ed6b966821462d374313256abbbe27f96ce11b2c