fix raise bug

PaddlePaddle · Aug 27, 2022 · 4ca5c41 · 4ca5c41
1 parent eca0483
commit 4ca5c41
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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,