torch.testing.assert_allclose
torch.testing.assert_close
* replace torch.testing.assert_allclose by torch.testing.assert_close * missing atol rtol