Fix _cat operator (#34591)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34591
Test Plan: Imported from OSS
Differential Revision: D20388000
Pulled By: VitalyFedyunin
fbshipit-source-id: 8ae7593dbddc1a96a03193a99afc9a4ce46203ad
