fix doc of take operator (apache#13947)

haohuanw · Jun 23, 2019 · c22a0cd · c22a0cd
1 parent 4822d5e
commit c22a0cd
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/operator/tensor/indexing_op.cc b/src/operator/tensor/indexing_op.cc
@@ -678,11 +678,14 @@ Examples::
   // In this case we will get rows 0 and 1, then 1 and 2 (calculated by wrapping around).
   // Along axis 1
 
-  take(x, [[0, 3], [-1, -2]], axis=1, mode='wrap') = [[[ 1.,  2.],
-                                                       [ 3.,  4.]],
+  take(x, [[0, 3], [-1, -2]], axis=1, mode='wrap') = [[[ 1.  2.]
+                                                       [ 2.  1.]]
 
-                                                      [[ 3.,  4.],
-                                                       [ 5.,  6.]]]
+                                                      [[ 3.  4.]
+                                                       [ 4.  3.]]
+
+                                                      [[ 5.  6.]
+                                                       [ 6.  5.]]]
 
 The storage type of ``take`` output depends upon the input storage type: