如何在TensorFlow中水平連接兩個張量？

Question

我有兩個張量形狀(100, 4)和(100, 2)。 我想在TensorFlow中執行一個級聯操作，與NumPy中的np.hstack類似，這樣輸出的形狀將是(100, 6)。是否有一個TensorFlow函數來做到這一點？

Answer 1

您可以使用tf.concat用於此目的如下：

sess=tf.Session() 
t1 = [[1, 2], [4, 5]] 
t2 = [[7, 8, 9], [10, 11, 12]] 
res=tf.concat(concat_dim=1,values=[t1, t2]) 
print(res.eval(session=sess)) 

這將打印

[[ 1 2 7 8 9] 
[ 4 5 10 11 12]]