diff options
| author | Stephan Hoyer <shoyer@google.com> | 2018-12-19 10:08:01 -0800 |
|---|---|---|
| committer | Stephan Hoyer <shoyer@google.com> | 2018-12-19 10:46:40 -0800 |
| commit | 45413455791c77465c5f33a5082053274eb18900 (patch) | |
| tree | 7362d739f4558b4aece09bbde333df5c219d8f1a /numpy/core/shape_base.py | |
| parent | f4ddc2b3251d4606cc227761933e991131425416 (diff) | |
| download | numpy-45413455791c77465c5f33a5082053274eb18900.tar.gz | |
ENH: refactor __array_function__ pure Python implementation
Diffstat (limited to 'numpy/core/shape_base.py')
| -rw-r--r-- | numpy/core/shape_base.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/numpy/core/shape_base.py b/numpy/core/shape_base.py index 0378d3c1f..f8332c362 100644 --- a/numpy/core/shape_base.py +++ b/numpy/core/shape_base.py @@ -342,10 +342,11 @@ def hstack(tup): def _stack_dispatcher(arrays, axis=None, out=None): arrays = _arrays_for_stack_dispatcher(arrays, stacklevel=6) - for a in arrays: - yield a if out is not None: - yield out + # optimize for the typical case where only arrays is provided + arrays = list(arrays) + arrays.append(out) + return arrays @array_function_dispatch(_stack_dispatcher) |