simplify `torch.meshgrid`'s shape computation (#62905)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62905
Reviewed By: mruberry
Differential Revision: D31021274
Pulled By: dagitses
fbshipit-source-id: c219389bdc543e9592f7b1c707acfbf752ee6f34