如何在tensorflow中导入/使用rfft?

时间:2017-03-21 11:30:38

标签: python tensorflow

最近,tensorflow增加了对rfft和irfft功能的支持。请参阅this github-issue,代码中添加了here。现在我想在我的模型中使用这些函数。我在码头工作环境中工作,所以我运行夜间图像(因为这是最近添加的)。我使用的是:'tensorflow / tensorflow:nightly-devel-gpu' 我查了一下,带有rfft ops的文件确实在这张图片中,位于:

/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/spectral_ops.py  

然后我尝试导入这些函数,这样的事情应该在我的脑海中起作用:

>>> from tensorflow import rfft
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ImportError: cannot import name rfft

然而,奇怪的是,这确实有效,没有错误:

>>> from tensorflow import fft  

当我打电话时:

import tensorflow as tf
dir(tf)

然后在长列表中显示fft和ifft函数,但不显示rfft和irfft函数。我假设我没有错误的版本,因为我可以检查我想要的代码是在图像上。另外,我不认为这是一个安装问题,因为我在tensorflow上直接使用干净的docker环境。
谁能看到我做错了什么?

1 个答案:

答案 0 :(得分:0)

解决了它,解决方案是使用:

from tensorflow.python.ops.spectral_ops import rfft