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

Question

我有 2 个形状为 (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]]