How to concatenate two tensors horizontally in TensorFlow?

后端 未结 3 815
终归单人心
终归单人心 2021-02-14 22:20

I have 2 tensors of shape (100, 4) and (100, 2). I would like to perform a concatenation operation, in TensorFlow, similar to np.hstack,

3条回答
  •  后悔当初
    2021-02-14 22:58

    You can use tf.concat for this purpose as follows:

    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))
    

    This prints

    [[ 1  2  7  8  9]
     [ 4  5 10 11 12]]
    

提交回复
热议问题