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,
(100, 4)
(100, 2)
np.hstack
You can use tf.concat for this purpose as follows:
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))
This prints
[[ 1 2 7 8 9] [ 4 5 10 11 12]]