如何向现有张量添加其他通道？

Question

我有一个大小为[3,3,3,64]的输入张量。我想在运行时向输入添加一个额外的通道，因此张量的形式为[3,3,4,64]。新的通道权重可以初始化为0。

我的问题是，如何插入新的频道数据以增加频道维度？

Answer 1

使用tf.concat

import tensorflow as tf

a = tf.ones([3, 3, 3, 64]) # your original stuff
b = tf.zeros([3, 3, 1, 64])

c = tf.concat([a, b], axis=2)
print c

将获得Tensor("concat_1:0", shape=(3, 3, 4, 64), dtype=float32)