在 TensorFlow 中将信息添加到张量中
Adding information into a tensor in TensorFlow
目前我有一个神经网络可以对图像进行卷积和池化。然而，就在我制作密集连接层之前，我想添加一些信息。目前，我使用 tf.reshape(image, [width * length * channels]) 将我的图像重塑为平面张量，但我想知道如何将几个 tf.float32 值附加到张量的末尾？
您可以使用 tf.reshape 和 -1:
重塑为矢量
tf.reshape(image, [-1])
并使用 tf.concat:
将新值附加为张量
tf.concat([image, new_val1, new_val2], 0)
这将 return 由输入张量串联产生的张量。
目前我有一个神经网络可以对图像进行卷积和池化。然而，就在我制作密集连接层之前，我想添加一些信息。目前，我使用 tf.reshape(image, [width * length * channels]) 将我的图像重塑为平面张量，但我想知道如何将几个 tf.float32 值附加到张量的末尾？
您可以使用 tf.reshape 和 -1:
tf.reshape(image, [-1])
并使用 tf.concat:
将新值附加为张量tf.concat([image, new_val1, new_val2], 0)
这将 return 由输入张量串联产生的张量。