首页 > 其他 > 详细

tf的concat问题

时间:2019-03-14 22:00:14      阅读:224      评论:0      收藏:0      [点我收藏+]

 

参考:https://xiulian.blog.csdn.net/article/details/86100641

 

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去掉两层。

 

tf的concat问题

原文:https://www.cnblogs.com/jianglinliu/p/10533639.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!