fix assert_allclose doesnt check shape (#47580)
Summary:
fix assert_allclose doesnt check shape
should fix https://github.com/pytorch/pytorch/issues/47449.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/47580
Reviewed By: samestep
Differential Revision: D24836399
Pulled By: walterddr
fbshipit-source-id: 943f8c83864bc01e1a782048c234e9592d2f1a25
