np.fft.fftshift не работает для python3.6 - PullRequest
0 голосов
/ 27 февраля 2019

Я пытаюсь использовать fftshift для объектов tenorflow.Он работает, когда я запускаю его на python2.7 на Mac, но когда я переключаюсь на python3.6 в Windows, он перестает работать.Вот простой тестовый код:

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)

Я проверил свою версию с ошибками, и это 1.16.1.

...