Replace torch.allClose with self.assertEqual (#39424)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39424
Reviewed By: Krovatkin
Differential Revision: D21854870
Pulled By: ailzhang
fbshipit-source-id: eb68f1775596e4c963169033444d6d6f4f818d4f