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

我有2个形状为(100, 4)和(100, 2)的张量。
我想在TensorFlow中执行串联操作，类似于NumPy中的np.hstack，以便输出的形状为(100, 6)。是否有一个TensorFlow函数可以做到这一点？

最佳答案

您可以为此使用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]]

关于tensorflow - 如何在TensorFlow中水平连接两个张量？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/43382216/