我有2个形状的张量 (100, 4) 和 (100, 2) . 我想在NensPy中的TensorFlow中执行类似于 np.hstack 的连接操作,以便输出形状为 (100, 6) . 是否有TensorFlow功能吗?
(100, 4)
(100, 2)
np.hstack
(100, 6)
您可以将 tf.concat 用于此目的,如下所示:
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]]
1 回答
您可以将
tf.concat
用于此目的,如下所示:这打印