如何在TensorFlow中水平连接两个张量？

Question

我有两个形状(100, 4)和张量的张量(100, 2).我想在TensorFlow中执行一个连接操作,类似于np.hstackNumPy,这样输出就会很好(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]]