tf的concat问题

最新推荐文章于 2022-11-18 18:07:12 发布

weixin_30437337

最新推荐文章于 2022-11-18 18:07:12 发布

阅读量74

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/jianglinliu/p/10533639.html

参考：1、https://xiulian.blog.youkuaiyun.com/article/details/86100641

　　　2、https://blog.youkuaiyun.com/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进行操作。

转载于:https://www.cnblogs.com/jianglinliu/p/10533639.html