tf.concat(values, axis)
concat沿着某一维度链接tensor, 需要理解tensor 的维度
t1 = [[[1, 1, 1],[2, 2, 2]],[[3, 3, 3],[4, 4, 4]]]
t2 = [[[5, 5, 5],[6, 6, 6]],[[7, 7, 7],[8, 8, 8]]]
第0维解释:
t1和t2的0维是两个tensor最外层的两个中括号, 所以 tf.concat(0, [t1,t2]), 就是将t1 和 t2的最外层中括号去掉,元素合在一起在最外层再加一个中括号即可
第1维解释:
t1和t2的1维中第一个元素分别是:[[1, 1, 1],[2, 2, 2]] 和 [[5, 5, 5],[6, 6, 6]], 分别把外层中括号去掉合在一起,即为[1, 1, 1], [2, 2, 2], [5, 5, 5], [6, 6, 6] 然后外层在加一个中括号即可[[1, 1, 1], [2, 2, 2], [5, 5, 5], [6, 6, 6]]
第2维解释:
t1和t2的2维中的第一个元素分别为[1, 1, 1] 和 [5, 5, 5],两者合并后即为 [1, 1, 1, 5, 5, 5]