1. ホーム
  2. tensorflow

[解決済み] TensorFlowで2つのテンソルを水平に連結する方法とは?

2022-02-25 14:08:44

質問

形状のテンソルを2つ持っている (100, 4)(100, 2) . TensorFlowで、次のような連結操作を行いたい。 np.hstack NumPyでは、出力は次のような形になります。 (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]]