更优雅的方式来沿轴复制张量

Question

我必须将我的一批图像复制到另一层3。 以下代码有效，但我想问一个更优雅的方法：

output = tf.stack([input[:,:,:,0], input[:,:,:,0], input[:,:,:,0]], axis=3)

感谢您的帮助!!!

Answer 1

您可以使用tf.tile。

arrayToStack = input[:,:,:,0]
shape = list(arrayToStack.shape)+[1]
output = tf.tile(arrayToStack.reshape(shape), [1,1,1,3])