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进行操作。

posted @ 2019-03-14 21:30  jianglin_liu  阅读(441)  评论(0编辑  收藏  举报