fix bug

Author: JiangZhaoh

python/mxnet/ndarray/numpy/_op.py

@@ -69,21 +69,17 @@ def append(arr, values, axis=None):
     delete : Delete elements from an array.
     Examples
     --------
-    >>> np.append([1, 2, 3], [[4, 5, 6], [7, 8, 9]])
-    array([1, 2, 3, ..., 7, 8, 9])
+    >>> np.append(np.array([1, 2, 3]), np.array([[4, 5, 6],[7, 8, 9]]))
+    array([1., 2., 3., 4., 5., 6., 7., 8., 9.])
     When `axis` is specified, `values` must have the correct shape.
-    >>> np.append([[1, 2, 3], [4, 5, 6]], [[7, 8, 9]], axis=0)
-    array([[1, 2, 3],
-           [4, 5, 6],
-           [7, 8, 9]])
-    >>> np.append([[1, 2, 3], [4, 5, 6]], [7, 8, 9], axis=0)
-    Traceback (most recent call last):
-        ...
-    ValueError: all the input arrays must have same number of dimensions
+    >>> np.append(np.array([[1, 2, 3], [4, 5, 6]]), np.array([[7, 8, 9]]), axis=0)
+    array([[1., 2., 3.],
+           [4., 5., 6.],
+           [7., 8., 9.]])
     """
     if axis is None:
-        arr = arr.ravel()
-        values = values.ravel()
+        arr = arr.flatten()
+        values = values.flatten()
         axis = 0
     return concatenate((arr, values), axis=axis)
 

python/mxnet/numpy/multiarray.py

@@ -6449,21 +6449,17 @@ def append(arr, values, axis=None):
     delete : Delete elements from an array.
     Examples
     --------
-    >>> np.append([1, 2, 3], [[4, 5, 6], [7, 8, 9]])
-    array([1, 2, 3, ..., 7, 8, 9])
+    >>> np.append(np.array([1, 2, 3]), np.array([[4, 5, 6],[7, 8, 9]]))
+    array([1., 2., 3., 4., 5., 6., 7., 8., 9.])
     When `axis` is specified, `values` must have the correct shape.
-    >>> np.append([[1, 2, 3], [4, 5, 6]], [[7, 8, 9]], axis=0)
-    array([[1, 2, 3],
-           [4, 5, 6],
-           [7, 8, 9]])
-    >>> np.append([[1, 2, 3], [4, 5, 6]], [7, 8, 9], axis=0)
-    Traceback (most recent call last):
-        ...
-    ValueError: all the input arrays must have same number of dimensions
+    >>> np.append(np.array([[1, 2, 3], [4, 5, 6]]), np.array([[7, 8, 9]]), axis=0)
+    array([[1., 2., 3.],
+           [4., 5., 6.],
+           [7., 8., 9.]])
     """
     if axis is None:
-        arr = arr.ravel()
-        values = values.ravel()
+        arr = arr.flatten()
+        values = values.flatten()
         axis = 0
     return concatenate((arr, values), axis=axis)
 

python/mxnet/symbol/numpy/_symbol.py

@@ -4583,21 +4583,17 @@ def append(arr, values, axis=None):
     delete : Delete elements from an array.
     Examples
     --------
-    >>> np.append([1, 2, 3], [[4, 5, 6], [7, 8, 9]])
-    array([1, 2, 3, ..., 7, 8, 9])
+    >>> np.append(np.array([1, 2, 3]), np.array([[4, 5, 6],[7, 8, 9]]))
+    array([1., 2., 3., 4., 5., 6., 7., 8., 9.])
     When `axis` is specified, `values` must have the correct shape.
-    >>> np.append([[1, 2, 3], [4, 5, 6]], [[7, 8, 9]], axis=0)
-    array([[1, 2, 3],
-           [4, 5, 6],
-           [7, 8, 9]])
-    >>> np.append([[1, 2, 3], [4, 5, 6]], [7, 8, 9], axis=0)
-    Traceback (most recent call last):
-        ...
-    ValueError: all the input arrays must have same number of dimensions
+    >>> np.append(np.array([[1, 2, 3], [4, 5, 6]]), np.array([[7, 8, 9]]), axis=0)
+    array([[1., 2., 3.],
+           [4., 5., 6.],
+           [7., 8., 9.]])
     """
     if axis is None:
-        arr = arr.ravel()
-        values = values.ravel()
+        arr = arr.flatten()
+        values = values.flatten()
         axis = 0
     return concatenate((arr, values), axis=axis)
 

tests/python/unittest/test_numpy_op.py

@@ -3555,12 +3555,13 @@ def hybrid_forward(self, F, a, b):
 
     def get_new_shape(shape, axis):
         shape_lst = list(shape)
-        shape_lst[axis] = random.randint(0, 3)
+        if axis is not None:
+            shape_lst[axis] = random.randint(0, 3)
         return tuple(shape_lst)
 
     for shape in [(0, 0), (2, 3)]:
         for hybridize in [True, False]:
-            for axis in range(2):
+            for axis in [0, 1, None]:
                 #test gluon
                 test_append = TestAppend(axis=axis)
                 if hybridize: