Fix tracing bugs where using `1 - x` in C++ would cause the size of 1 to get hardcoded. (#20932)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20932
ghimport-source-id: f0a7f12ffd77aec063a088b18c6b1d108c712df8
Differential Revision: D15501251
Pulled By: zdevito
fbshipit-source-id: 91e6e5944d2663b673afde45fc6eed22f31101c4