diff options
Diffstat (limited to 'numpy/fft/helper.py')
| -rw-r--r-- | numpy/fft/helper.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/fft/helper.py b/numpy/fft/helper.py index 0832bc5a4..0856d6759 100644 --- a/numpy/fft/helper.py +++ b/numpy/fft/helper.py @@ -64,7 +64,7 @@ def fftshift(x, axes=None): """ tmp = asarray(x) - ndim = len(tmp.shape) + ndim = tmp.ndim if axes is None: axes = list(range(ndim)) elif isinstance(axes, integer_types): @@ -113,7 +113,7 @@ def ifftshift(x, axes=None): """ tmp = asarray(x) - ndim = len(tmp.shape) + ndim = tmp.ndim if axes is None: axes = list(range(ndim)) elif isinstance(axes, integer_types): |