前言:tensorflow1.4.0
定义:将张量连接到一个维度上。
代码实现:
import tensorflow as tf
t1 = [[1, 2, 3], [4, 5, 6],[7,8,9]]
t2 = [[10,11,12],[13,14,15],[16,17,18]]
with tf.Session() as sess:
print(sess.run(tf.concat([t1, t2], 0))) #按行合并,将矩阵中的每一行元素为一个单元进行逐行合并。
print(sess.run(tf.concat([t1, t2], 1))) #按列合并,将矩阵中的每一列元素为一个单元进行逐列合并。
结果:
[[ 1 2 3]
[ 4 5 6]
[ 7 8 9]
[10 11 12]
[13 14 15]
[16 17 18]]
[[ 1 2 3 10 11 12]
[ 4 5 6 13 14 15]
[ 7 8 9 16 17 18]]