np.fft.fftshift不适用于python3.6

Question

我正在尝试对tensorflow对象使用fftshift。当我在Mac上的python2.7上运行它时，它可以工作，但是当我在Windows上切换到python3.6时，它停止工作了。 这是一个简单的测试代码：

import numpy as np
import tensorflow as tf
a = np.random.rand(10,10)
b = tf.fft2d(a)
c = np.fft.fftshift(b)
error message:
np.fft.fftshift(tf.fft2d(a))
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\Users\kathy\AppData\Local\Programs\Python\Python36\lib\site-packages\numpy\fft\helper.py", line 81, in fftshift
    return roll(x, shift, axes)
  File "C:\Users\kathy\AppData\Local\Programs\Python\Python36\lib\site-packages\numpy\core\numeric.py", line 1485, in roll
    arr_index, res_index = zip(*indices)
ValueError: not enough values to unpack (expected 2, got 0)

我检查了numpy版本，它是1.16.1。