summaryrefslogtreecommitdiff
path: root/numpy/fft/fftpack.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/fft/fftpack.py')
-rw-r--r--numpy/fft/fftpack.py15
1 files changed, 10 insertions, 5 deletions
diff --git a/numpy/fft/fftpack.py b/numpy/fft/fftpack.py
index 236a18d7a..706fcdd2f 100644
--- a/numpy/fft/fftpack.py
+++ b/numpy/fft/fftpack.py
@@ -829,11 +829,16 @@ def fft2(a, s=None, axes=(-2, -1)):
--------
>>> a = np.mgrid[:5, :5][0]
>>> np.fft.fft2(a)
- array([[ 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
- [ 5.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
- [ 10.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
- [ 15.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
- [ 20.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j]])
+ array([[ 50.0 +0.j , 0.0 +0.j , 0.0 +0.j ,
+ 0.0 +0.j , 0.0 +0.j ],
+ [-12.5+17.20477401j, 0.0 +0.j , 0.0 +0.j ,
+ 0.0 +0.j , 0.0 +0.j ],
+ [-12.5 +4.0614962j , 0.0 +0.j , 0.0 +0.j ,
+ 0.0 +0.j , 0.0 +0.j ],
+ [-12.5 -4.0614962j , 0.0 +0.j , 0.0 +0.j ,
+ 0.0 +0.j , 0.0 +0.j ],
+ [-12.5-17.20477401j, 0.0 +0.j , 0.0 +0.j ,
+ 0.0 +0.j , 0.0 +0.j ]])
"""