From 4ca5c417c69ab0042702ecd13fd5d128571d9717 Mon Sep 17 00:00:00 2001 From: S-HuaBomb Date: Sat, 27 Aug 2022 12:38:36 +0800 Subject: [PATCH] fix raise bug --- python/paddle/tensor/math.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/python/paddle/tensor/math.py b/python/paddle/tensor/math.py index 1d15475572f65..9dd1ab36fd0f3 100644 --- a/python/paddle/tensor/math.py +++ b/python/paddle/tensor/math.py @@ -4863,9 +4863,7 @@ def take(x, index, mode='raise', name=None): if mode == 'raise': # This processing enables 'take' to handle negative indexes within the correct range. - # Negative indexes can be enabled, - # but out-of-range indexes will report an error in the following paddle.index_select - index_1d = paddle.where(index_1d < 0, index_1d % max_index, index_1d) + index_1d = paddle.where(index_1d < 0, index_1d + max_index, index_1d) elif mode == 'wrap': # The out of range indices are constrained by taking the remainder. index_1d = paddle.where(index_1d < 0,