fix doc for sort and argsort (#15317)

apache · Jul 8, 2019 · 1ae73de · 1ae73de
1 parent 091fece
commit 1ae73de
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/operator/tensor/ordering_op.cc b/src/operator/tensor/ordering_op.cc
@@ -114,7 +114,7 @@ Examples::
              [ 1.,  3.]]
 
   // flattens and then sorts
-  sort(x) = [ 1.,  1.,  3.,  4.]
+  sort(x, axis=None) = [ 1.,  1.,  3.,  4.]
 
   // sorts along the first axis
   sort(x, axis=0) = [[ 1.,  1.],
@@ -173,7 +173,7 @@ Examples::
                         [ 0.,  1.,  0.]]
 
   // flatten and then sort
-  argsort(x) = [ 3.,  1.,  5.,  0.,  4.,  2.]
+  argsort(x, axis=None) = [ 3.,  1.,  5.,  0.,  4.,  2.]
 )code" ADD_FILELINE)
 .set_num_inputs(1)
 .set_num_outputs(1)