pytorch

e3403ff4 - square support

square support Signed-off-by: Edward Z. Yang <ezyangfb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/77682 Approved by: https://github.com/ngimel, https://github.com/mruberry