tf的concat问题
参考:1、https://xiulian.blog.csdn.net/article/details/86100641
2、https://blog.csdn.net/fangjian1204/article/details/53055219
import tensorflow as tf a = [1, 2, 3] b = [4, 5, 6] c = tf.concat([a, b], axis=0) sess = tf.InteractiveSession() #建立交互式会话 print(sess.run(c)) #输出[1 2 3 4 5 6]
a = [[[1,1],[2,2]],[[3,3],[4,4]]] b = [[[5,5]],[[6,6]]] c = tf.concat([a, b], axis=1) print(sess.run(c)) #输出[[[1 1] [2 2] [5 5]] [[3 3] [4 4] [6 6]]]
根据tf.concat的axis参数来去数组的[],0表示去掉最外面的一层,1去掉两层。这个可以判断维度是否相同,即可不可以进行操作。
参考2介绍了多维的数据如何按axis进行操作。