[ONNX] Update assign output shape for nested structure and dict output (#52893) (#53311)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53311
Fixes dict output & nested tuple.
Test Plan: Imported from OSS
Reviewed By: pbelevich, malfet
Differential Revision: D26922426
Pulled By: SplitInfinity
fbshipit-source-id: c2c6b71c8d978b990181e0b025626dbf6ef2199e
