Replace use of np.allclose in tests. (#34287)
Summary:
fixes https://github.com/pytorch/pytorch/issues/34096
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34287
Differential Revision: D21735525
Pulled By: ailzhang
fbshipit-source-id: 611da17cfc5a3fee77d482abccf8f9854f504263
