use torch.testing.assert_close in test (#1016)

use torch.testing.assert_close in test (#1016) #3625