Skip to content

Commit 2897f07

Browse files
committed
Add take_along_axis to choose docstring and fix a typo
1 parent bbbf2e8 commit 2897f07

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dpnp/dpnp_iface_indexing.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,11 +220,12 @@ def choose(a, choices, out=None, mode="wrap"):
220220
See also
221221
--------
222222
:obj:`dpnp.ndarray.choose` : Equivalent method.
223+
:obj:`dpnp.take_along_axis` : Preferable if choices is an array.
223224
224225
Examples
225226
--------
226227
>>> import dpnp as np
227-
>>> choices = np.array[[0, 1, 2, 3], [10, 11, 12, 13],
228+
>>> choices = np.array([[0, 1, 2, 3], [10, 11, 12, 13],
228229
... [20, 21, 22, 23], [30, 31, 32, 33]])
229230
>>> np.choose(np.array([2, 3, 1, 0]), choices
230231
... # the first element of the result will be the first element of the

0 commit comments

Comments
 (0)