fix bug:When axes in paddle.slice is a tuple, an error occurs. (#35267)

* fix bug:When axes in paddle.sile is a tuple, an error occurs. * polish code.
PaddlePaddle · Sep 1, 2021 · b53887f · b53887f
1 parent 3c21f26
commit b53887f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
@@ -10977,6 +10977,7 @@ def slice(input, axes, starts, ends):
         ends_tensor = None
 
         if isinstance(axes, (list, tuple)):
+            axes = list(axes)
             if len(axes) == 0:
                 raise ValueError(
                     "Input axes should not be an empty list/tuple.")

diff --git a/python/paddle/fluid/tests/unittests/test_slice_op.py b/python/paddle/fluid/tests/unittests/test_slice_op.py
@@ -705,7 +705,7 @@ def test_axis_less_than_zero(self):
             np_slice = x_arr[:, :, 0:1]
             self.assertTrue(np.array_equal(pp_slice, np_slice))
 
-            pp_slice = paddle.slice(x, [-100, ], [0], [1])
+            pp_slice = paddle.slice(x, (-100, ), [0], [1])
             np_slice = x_arr[0:1]
             self.assertTrue(np.array_equal(pp_slice, np_slice))