scatter/gather - check that inputs are of the same dimensionality (#41672)
Summary:
As per title.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41672
Reviewed By: malfet, ngimel
Differential Revision: D22678302
Pulled By: gchanan
fbshipit-source-id: 95a1bde81e660b8963e5914d5348fd4fbff1338e