Working on component-wise transformations that mimic `torch.cat` and `torch.stack` (#11868)
Summary:
As discussed in #11755 .
Pull Request resolved: https://github.com/pytorch/pytorch/pull/11868
Differential Revision: D10032248
Pulled By: ezyang
fbshipit-source-id: d3a81c19f65a3e716f7f1cfc0a42b86c32fc484c
