如何在输入通道上单独执行卷积?

时间:2017-05-15 07:52:17

标签: tensorflow

是否可以单独对输入通道执行卷积?由于有两个输入通道具有深度2.如果我将过滤器设置为[2,2,1,1],则会给出错误。那么如何在两个输入通道上单独执行卷积?

input = tf.Variable(tf.random_normal([1,4,4,2]))
filter = tf.Variable(tf.random_normal([2,2,2,1]))
op = tf.nn.conv2d(input, filter, strides=[1, 2, 2, 1], padding='SAME')

1 个答案:

答案 0 :(得分:0)

  

如何在输入通道上单独执行卷积?

tf.nn.depthwise_conv2d

“对每个输入通道应用不同的过滤器(每个通道从1个通道扩展到channel_multiplier通道),然后将结果连接在一起。”