pytorch

157d478a - Fix omission of shape in size check in index.

Fix omission of shape in size check in index. Signed-off-by: Edward Z. Yang <ezyangfb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/78897 Approved by: https://github.com/Lezcano, https://github.com/anjali411