From a2ea6f63a3cb20f8bf5f6494eff0727b7f14bc3d Mon Sep 17 00:00:00 2001 From: Alex Li Date: Tue, 14 Aug 2018 21:33:36 -0700 Subject: [PATCH] update ndarray stack Doc for #11925 (#12015) * update ndarray stack Doc --- python/mxnet/ndarray_doc.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) 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