【PaddlePaddle Hackathon 3 No.16】为 Paddle 新增 API paddle.take

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,