Tensorflow乘法广播(求和前的卷积)

时间:2018-10-14 15:51:43

标签: python tensorflow image-processing filter convolution

我正在寻找一个张量流API,该API需要一个过滤器并经过一个张量并将它们相乘,然后为每个元素提供一个矢量。

让我们更加清楚。我有2D灰度图像矩阵。我有一个2D滤镜(例如3x3),我将滤镜跨过图像并为每个像素获取9个长度的矢量,以使每个矢量的元素都是对应的滤镜元素和像素邻居相乘。换句话说,求和之前我需要conv2d的结果。该向量的总和就是滤波器和相应像素的卷积。

我找不到用于此操作的任何标准API。我可以编写一个遍历元素的代码,但是效率不高。有谁知道如何做到这一点?

0 个答案:

没有答案