first fix the UT

Author: zoooo0820

test/legacy_test/test_zero_dim_tensor.py

@@ -830,6 +830,7 @@ def test_setitem(self):
         np.testing.assert_allclose(out[1, 2, 3, 4], np.array(10))
         self.assertEqual(x.grad.shape, [2, 3, 4, 5])
         x_grad_expected = np.ones((2, 3, 4, 5)) * 2
+        x_grad_expected[1, 2, 3, 4] = 0
         np.testing.assert_allclose(x.grad, x_grad_expected)
 
         # case2: 0-D Tensor indice in some axis
@@ -847,6 +848,7 @@ def test_setitem(self):
         self.assertEqual(out.shape, x.shape)
         np.testing.assert_allclose(out[1, 1], np.ones((4, 5)) * 0.5)
         x_grad_expected = np.ones((2, 3, 4, 5))
+        x_grad_expected[1, 1] = 0
         np.testing.assert_allclose(x.grad, x_grad_expected)
 
         # case3：0-D Tensor indice in some axis, value is a Tensor

test/xpu/test_zero_dim_tensor_xpu.py

@@ -441,6 +441,7 @@ def test_setitem(self):
         np.testing.assert_allclose(out[1, 2, 3, 4], np.array(10))
         self.assertEqual(x.grad.shape, [2, 3, 4, 5])
         x_grad_expected = np.ones((2, 3, 4, 5)) * 2
+        x_grad_expected[1, 2, 3, 4] = 0
         np.testing.assert_allclose(x.grad, x_grad_expected)
 
         # case2: 0-D Tensor indice in some axis