From cfc9bf7677e729994264644781565ef015751c48 Mon Sep 17 00:00:00 2001 From: Weilong Wu Date: Thu, 4 Aug 2022 15:17:19 +0800 Subject: [PATCH] [Eager] fix slice's input mistake (#44855) * [Eager] fix slice's input mistake * add tests for slice --- python/paddle/fluid/tests/unittests/test_slice_op.py | 2 +- python/paddle/tensor/manipulation.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_slice_op.py b/python/paddle/fluid/tests/unittests/test_slice_op.py index cdffb6fb37f..41a4d35d906 100644 --- a/python/paddle/fluid/tests/unittests/test_slice_op.py +++ b/python/paddle/fluid/tests/unittests/test_slice_op.py @@ -635,7 +635,7 @@ class TestSliceApiEager(unittest.TestCase): axes=axes, starts=paddle.to_tensor(starts), ends=paddle.to_tensor(ends)) - + self.assertTrue(np.array_equal(a_1.numpy(), a_2.numpy())) a_1.backward() grad_truth = paddle.zeros_like(a) grad_truth[-3:3, 0:2, 2:4] = 1 diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py index a280baa17b1..a821d6f72fc 100755 --- a/python/paddle/tensor/manipulation.py +++ b/python/paddle/tensor/manipulation.py @@ -208,7 +208,7 @@ def slice(input, axes, starts, ends): if isinstance(item, tmp_tensor_type) else item for item in ends ] elif isinstance(ends, tmp_tensor_type): - etensor_t = ends.numpy() + tensor_t = ends.numpy() ends = [ele for ele in tensor_t] infer_flags = list(-1 for i in range(len(axes))) -- GitLab