Tensorflow比scipy慢,并且嵌套for循环

时间:2018-04-03 00:28:30

标签: tensorflow scipy

我的输入张量 11x11x4 和2个形状 5x5x4 的内核。我试图关联(卷积)他们。我使用scipy.signal.correlatetensorflow.nn.conv2d和嵌套的for循环,使用3D numpy数组的元素乘法。此操作所需的时间如下: 的时间(SciPy的)LT;时间(嵌套用于循环)<<时间(tensorflow)即可。相应的代码和时序图可以在jupyter notebook中找到。请滚动至时序分析部分。令人惊讶的是,张量流比嵌套for循环更糟糕。我在这里做错了吗?或者我比较苹果和橘子?如果是的话,我应该使用什么操作才能建立公平竞争?我没有使用GPU。

0 个答案:

没有答案