[testing] Support input samples where `self` is broadcasted. (#53014)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/50747
Reference https://github.com/pytorch/pytorch/issues/50006
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53014
Reviewed By: SplitInfinity, ngimel
Differential Revision: D27615320
Pulled By: mruberry
fbshipit-source-id: a48bccf06aef1ee8f66a89e6b2bbe79736700b2b
