Tensorflow concat 2D寸法テンソルから3D寸法テンソル

Question

これはむしろ簡単な質問です。 3Dテンソルを持っています。それは[2,2,3]と言うことができます。私は2D形状の別のテンソルを持っています[2,2]。しかし、両方の要素の最初の2つの次元は一致しています。私はconcatそれらに2Dテンソルが3次元の3Dテンソルに追加されるようにしたいと思います。すなわち [2,2,4]です。

これを達成する方法がわかりません。私はtf.concatとtf.stackを使ってみましたが、両方のテンソルを同じランクにする必要があります。

Answer 1

あなたはtf.concat()続い正しいランクに小さいテンソルを変換するtf.expand_dims()を使用することができます。

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]