Tensorflow Concat 2D尺寸张量到3D尺寸张量

Question

这是一个简单的问题。 我有一个3D形状的张量。 可以说是[2,2,3] 。 我还有一个2D形状的张量[2,2] 。 但是这两个元素的前两个维度是匹配的。 我想对它们进行concat ，以便将2D张量添加到三维的3D张量中。 即[2,2,4] 。

我不确定如何实现这一目标。 我尝试使用tf.concat和tf.stack但这需要两个张量都处于同一等级。

Answer 1

您可以使用tf.expand_dims()将较小的张量转换为正确的秩，然后使用tf.concat() ：

tensor_3d = tf.placeholder(tf.float32, shape=[2, 2, 3])
tensor_2d = tf.placeholder(tf.float32, shape=[2, 2])

tensor_2d_as_3d = tf.expand_dims(tensor_2d, 2)  # shape: [2, 2, 1]

result = tf.concat([tensor_3d, tensor_2d_as_3d], 2)  # shape: [2, 2, 4]