Skip to content

Commit

Permalink
update ndarray stack Doc for apache#11925 (apache#12015)
Browse files Browse the repository at this point in the history
* update ndarray stack Doc
  • Loading branch information
liyujiel authored and nswamy committed Aug 15, 2018
1 parent d12d1c6 commit a2ea6f6
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions python/mxnet/ndarray_doc.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit a2ea6f6

Please sign in to comment.