[Shape Inference] Infer input(1) from input(0) in elementwise ops (#36498)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/36498
Test Plan: unit test
Reviewed By: yinghai
Differential Revision: D20996757
fbshipit-source-id: 35e5e0bed63d2ba8a0699a8a78bff6f78e7af23c