Add squeeze, unsqueeze, transpose shape functins (#63099)
authorElias Ellison <eellison@devfair044.h1.fair>
Wed, 8 Sep 2021 01:19:14 +0000 (18:19 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 8 Sep 2021 01:22:19 +0000 (18:22 -0700)
commitc8a608b197fabb9f572e46df3459a8eeeafa5411
tree300d3e36d45056c3b53bc7f11eb073069721b7d9
parenta39f3c68b79a2bd0953e45e28b3540519424e7f0
Add squeeze, unsqueeze, transpose shape functins (#63099)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63099

These are checked by OpInfos, which represent all of the inputs and semantics of the operators so it should be an easy stamp

Test Plan: Imported from OSS

Reviewed By: desertfire, astaff

Differential Revision: D30347514

Pulled By: eellison

fbshipit-source-id: 37b4c9ecd8c222cc12bf39166181464b43218830
test/test_ops.py
torch/csrc/jit/runtime/symbolic_shape_registry.cpp
torch/testing/_internal/common_methods_invocations.py