torch.square : OpInfo and minor fixes (#52551)
Summary:
Reference: https://github.com/pytorch/pytorch/issues/42515
Add `out` variant to be consistent with Unary Ops.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/52551
Reviewed By: heitorschueroff
Differential Revision: D27233482
Pulled By: mruberry
fbshipit-source-id: fef6f241849a12c46028bd1aad8f5ecc1dc65ea1
