use `torch.testing.assert_equal` in `TestCase.assertEqual` (#67796)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/67796
Supersedes #58981.
cc mruberry
Test Plan: Imported from OSS
Reviewed By: ngimel
Differential Revision: D33542994
Pulled By: mruberry
fbshipit-source-id: 527099f5fdc154fd95ee48cd19f0a85eeec43443
(cherry picked from commit 1a58915e2cfde5c48ad77198a917872a03fd1b72)