Tensorflow 如何向现有张量添加额外通道？

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

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

使用

将获得

张量（“concat_1:0”，shape=（3,3,4,64），dtype=float32）

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