diff --git a/python/paddle/fluid/tests/unittests/test_fill_constant_op.py b/python/paddle/fluid/tests/unittests/test_fill_constant_op.py index b73711b19d613c12defc92f17c1396989415018e..537cabd5d09b7d72c55a9d305839780ab58544f3 100644 --- a/python/paddle/fluid/tests/unittests/test_fill_constant_op.py +++ b/python/paddle/fluid/tests/unittests/test_fill_constant_op.py @@ -62,8 +62,10 @@ class TestFillConstantOpWithSelectedRows(OpTest): fill_constant_op.run(scope, place) # get result from Out - result_array = np.array(out) - self.assertEqual(result_array, np.full((123, 92), 3.8)) + result_array = np.array(out.get_tensor()) + full_array = np.full((123, 92), 3.8, 'float32') + + self.assertTrue(np.array_equal(result_array, full_array)) def test_fill_constant_with_selected_rows(self): places = [core.CPUPlace()]