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

2024-04-27 05:20:00 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在寻找一个张量流API,它需要一个过滤器,通过一个张量,将它们相乘,并为每个元素提供一个向量。你知道吗

让我们说清楚一点。我有二维矩阵的灰度图像。我有一个2D滤波器(例如3x3),我将在图像上滚动我的滤波器,得到每个像素的一个9长度的向量,这样每个向量的元素就是对应的滤波器元素和像素邻居乘法。换句话说,在求和之前我需要conv2d的结果。这个向量的和就是滤波器和相应像素的卷积。你知道吗

我找不到任何用于此操作的标准API。我可以编写一个在元素上循环的代码,但效率不高。有人知道怎么做吗?你知道吗


Tags: 代码图像api元素过滤器标准矩阵像素