Fixed cat transform to work with event_dim > 0

[vikigenius]

Fixes #44530

As explained in the issue description, CatTransform does not work with event_dim > 0.
This PR fixes this. If this gets approved I am hoping to do the same for StackTransform as well.

@fritzo Can you take a look at this ?

[facebook-github-bot]

Hi @vikigenius!

Thank you for your pull request and welcome to our community. We require contributors to sign our Contributor License Agreement, and we don't seem to have you on file.

In order for us to review and merge your code, please sign at https://code.facebook.com/cla. If you are contributing on behalf of someone else (eg your employer), the individual CLA may not be sufficient and your employer may need to sign the corporate CLA.

If you have received this in error or have any questions, please contact us at cla@fb.com. Thanks!

[dr-ci]

💊 CI failures summary and remediations

As of commit 405dc58 (more details on the Dr. CI page):

---

✅ None of the CI failures appear to be your fault 💚

---

• 3/3 broken upstream at merge base fc0a3a1 on Dec 09 from 12:34pm to 9:07pm

---

🚧 3 fixed upstream failures:

These were probably caused by upstream breakages that were already fixed.

Please rebase on the viable/strict branch (expand for instructions)

If your commit is older than viable/strict, run these commands:

git fetch https://github.com/pytorch/pytorch viable/strict
git rebase FETCH_HEAD

Check out the recency history of this "viable master" tracking branch.

• pytorch_linux_xenial_cuda10_2_cudnn7_py3_gcc7_test2 on Dec 09 from 12:40pm to 7:07pm (fc0a3a1 - dfa3808)
  • 🔁 rerun
• pytorch_windows_vs2019_py36_cuda10.1_test2 on Dec 09 from 12:34pm to 4:54pm (f5e9ffb - e5a98c5)
  • 🔁 rerun
• pytorch_linux_xenial_cuda10_2_cudnn7_py3_gcc7_test1 on Dec 09 from 12:40pm to 9:07pm (fc0a3a1 - a6fa3b2)
  • 🔁 rerun

---

This comment was automatically generated by Dr. CI (expand for details).

Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions to the (internal) Dr. CI Users group.

This comment has been revised 11 times.

[fritzo]

LGTM, thanks for coding up the solution!

[fritzo]

test failures appear unrelated to me.

[fritzo]

@pytorchbot merge this please

[facebook-github-bot]

@ezyang has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

[facebook-github-bot]

@ezyang merged this pull request in 1e2d1d7.