最近,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环境。
谁能看到我做错了什么?
答案 0 :(得分:0)
解决了它,解决方案是使用:
from tensorflow.python.ops.spectral_ops import rfft