将要素映射与自身进行卷积

Question

我使用TensorFlow 1.12

我想拿一批特征地图[B,H,W,C]，并想卷积每个通道与它自己。

对于tf.map_fn，这可能是可能的，但我希望尽可能地保持这些操作的矢量化。

实现它的最佳矢量化方法是什么？

Answer 1

每个通道都是一个图像。使用convolution theorem在傅里叶域中最有效地实现图像与另一图像之间的卷积。它指出，一幅图像与另一幅图像的卷积与其傅里叶变换的点积的逆傅里叶变换相同。将其分解为几个步骤：

傅里叶变换的点积，pad the images with zeros

• Fourier transform，images.

• Calculate，images.

• Calculate，transforms.

• Inverse Fourier transform.

，，

这两个图像是相同的是一个特例。