FFT卷积使用4D滤镜对4D图像进行卷积

Question

我只想用纯NumPy在Python上创建CNN网络。为了优化网络中的卷积时间（实际上是相关性），我想尝试使用基于FFT的卷积。 需要卷积（相关）的数据是具有形状的4D图像张量 [batch_size, width, height, channels]和4D滤波器张量 [filter_width, filter_height, in_channel, out_channel]。 我读了很多有关基于FFT的卷积的文章，但它们并非以我的方式做。 因此，我需要您的帮助。