Question

我正在尝试在tensorflow中使用tf.image_summary函数。我正在尝试可视化卷积层的过滤器。过滤器定义为tf.Variable(tf.constant(0.1, shape=[5, 5, 16, 32]))。

但是在这里，因为我只想看到最终的过滤器，我想找到一种方法来获得大小为[5,5,32]的过滤器，只需取16的维度的第一个索引。如果我使用[：，：，0，：]然后我假设我会得到一个[5,5,1,32]过滤器而不是我想要的[5,5,32]。

我该怎么办？

Answer 1

所以tf.image_summary作为输入接受批处理，但是它在颜色通道方面需要1,3或4个。

所以你必须传递给tf.image_summary这样的东西：

for i in range(int(math.floor(filter.get_shape()[4]/3))):
    tf.image_summary(filter[:,:,:,i:i+3])