在TensorFlow中将信息添加到张量中

Question

目前我有一个神经网络可以卷积和汇集图像。但是，在我建立密集连接层之前，我想添加一些信息。目前，我使用tf.reshape(image, [width * length * channels])将我的图像重新整形为平坦的张量，但我想知道如何在张量的末尾附加几个tf.float32值？

Answer 1

您可以使用tf.reshape和-1

重塑为矢量

tf.reshape(image, [-1])

并使用tf.concat将新值附加为张量：

tf.concat(0, [image, new_val1, new_val2])

这将返回由输入张量连接产生的张量。