replace `self.assertTrue(torch.allclose(..))` with `self.assertEqual(…)` (#63637)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/63565
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63637
Reviewed By: malfet
Differential Revision: D30541266
Pulled By: mruberry
fbshipit-source-id: ab461949782c6908a589ea098fcfcf5c3e081ee6