diff options
| author | Javier Dehesa <javidcf@gmail.com> | 2019-05-21 16:18:24 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-05-21 16:18:24 +0100 |
| commit | 62301b08a4cd2a3c3cd214f89faa72f12b68dbf3 (patch) | |
| tree | a3ec1a991f73d12b97118f1185c516fd3efa9704 /numpy/lib/shape_base.py | |
| parent | 9608ac937e447380e730b49b146599c923b6a23b (diff) | |
| download | numpy-62301b08a4cd2a3c3cd214f89faa72f12b68dbf3.tar.gz | |
Fixed minor doc error in take_along_axis
Fixes #13598
Diffstat (limited to 'numpy/lib/shape_base.py')
| -rw-r--r-- | numpy/lib/shape_base.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/numpy/lib/shape_base.py b/numpy/lib/shape_base.py index 1b66f3a3e..f2517b474 100644 --- a/numpy/lib/shape_base.py +++ b/numpy/lib/shape_base.py @@ -94,7 +94,7 @@ def take_along_axis(arr, indices, axis): Ni, M, Nk = a.shape[:axis], a.shape[axis], a.shape[axis+1:] J = indices.shape[axis] # Need not equal M - out = np.empty(Nk + (J,) + Nk) + out = np.empty(Ni + (J,) + Nk) for ii in ndindex(Ni): for kk in ndindex(Nk): |