1. 程式人生 > >Tensorflow tf.concat

Tensorflow tf.concat

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]