[tensorexpr] Add support for aten::stack (#73801)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/73801
Test Plan: Imported from OSS
Reviewed By: ZolotukhinM
Differential Revision: D34647822
Pulled By: huiguoo
fbshipit-source-id: 3b863c71886c7c6616b16f5d3313079714c8b82a
(cherry picked from commit c71778cf6a5724d26b671bf3ee0478add24990e8)
