Handle None indexing in TorchScript (#18615)
authorThomas Viehmann <tv.code@beamnet.de>
Mon, 8 Apr 2019 20:35:34 +0000 (13:35 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Mon, 8 Apr 2019 20:44:49 +0000 (13:44 -0700)
commit026a9c6bf2e17c724f011adb6ed9c051fe7e4b65
tree703da75afd0232075cb706ed6b87ced5df4fbffc
parent239de1623d07b3b88900ba54a37d43d66e3bffa5
Handle None indexing in TorchScript (#18615)

Summary:
t[None], t[None, 1:, None] and friends for unsqueezing

Fixes: #12810
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18615

Differential Revision: D14837039

Pulled By: wanchaol

fbshipit-source-id: ab3862c41629f087b0a46b7c59c93dac4018e6fe
test/test_jit.py
torch/csrc/jit/script/compiler.cpp