ArrayFire中的卷积使用快速傅立叶变换吗？

Question

我期待着加快对大型医学图像(512 x 512 x 1000倍)上高斯核的导数(2/3阶)的卷积，在我们的开源工具包中。我们目前通过FFT进行这种卷积运算。

在一位朋友建议我使用ArrayFire之后，在阅读了这个帖子之后，我想看看是否可以采用这个工具包。似乎是一个伟大的努力，使我们能够处理多个后端，虽然我目前只对CUDA感兴趣，因为这是我手头的东西。

我在论坛上读到这个帖子，它说ArrayFire中的卷积在特定内核大小之后切换到频域。我查看了cuda文件convolve.cu，但没有在ArrayFire或任何cuFFT文件中找到对fft的调用。我是不是遗漏了什么？

接下来，我想直接在频域构造高斯核的导数，用图像FFT进行乘法，并将其带回来。但是我想比较空间中创建卷积内核和freq域之间的速度提高。而且，ArrayFire在3D中似乎没有高斯核。

Answer 1

您的观察是正确的，我们还没有为更大的内核大小实现fft回退。但是，我们正在尝试在接下来的几周内添加此功能。您在应用程序中使用的内核大小是多少？

我们目前还没有实现三维高斯核函数。但是我相信它可以很容易地在ArrayFire中使用启用JIT的元素操作来实现。有关如何使用元素级操作的示例，请查看这里。