函数原型:
tf.concat([tensor1, tensor2, tensor3,...], axis)
tensorflow中用于拼接张量。
举例:
t1 = [[1, 2, 3], [4, 5, 6]]
t2 = [[7, 8, 9], [10, 11, 12]]
tf.concat([t1, t2], 0) # [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]]
tf.concat([t1, t2], 1) # [[1, 2, 3, 7, 8, 9], [4, 5, 6, 10, 11, 12]]
# tensor t3 with shape [2, 3]
# tensor t4 with shape [2, 3]
tf.shape(tf.concat([t3, t4], 0)) # [4, 3]
tf.shape(tf.concat([t3, t4], 1)) # [2, 6]
感觉从第二个例子中更好理解,axis = 0就是将shape = [2,3]的t3、t4从第一个维度2拼接起来变成了4,第二个维度3则不变。axis = 1就是将shape = [2,3]的t3、t4从第二个维度3拼接起来变成了6,第一个维度2则不变。