diff --git a/python/mxnet/ndarray_doc.py b/python/mxnet/ndarray_doc.py index 0c51036d8208..9d6258a89a3d 100644 --- a/python/mxnet/ndarray_doc.py +++ b/python/mxnet/ndarray_doc.py @@ -105,6 +105,21 @@ class BroadcastToDoc(NDArrayDoc): (2L, 2L, 2L, 3L) """ +class StackDoc(NDArrayDoc): + """ + Example + -------- + Join a sequence of arrays along a new axis. + >>> x = mx.nd.array([1, 2]) + >>> y = mx.nd.array([3, 4]) + >>> stack(x, y) + [[1, 2], + [3, 4]] + >>> stack(x, y, axis=1) + [[1, 3], + [2, 4]] + """ + class CustomDoc(NDArrayDoc): """ Example