Fix concat dimension check bug (#17343)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17343
See [post](https://fb.workplace.com/groups/1405155842844877/permalink/2630764056950710/)
Reviewed By: dzhulgakov
Differential Revision: D14163001
fbshipit-source-id: 038f15d6a58b3bc31910e7bfa47c335e25739f12
