tf.concat feature map张量拼接

实验中时常会遇到张量拼接的问题,需要用到tf.concat。

对于两个1024的feature,需要先扩展维度变成1*1024,再拼接,代码如下:

    tf.expand_dims(y11,1)
    tf.expand_dims(y13,1)
    y14=tf.concat([y11,y13],1)

其中y11和y12都是1024,拼接后的y14是1*2048。

猜你喜欢

转载自blog.csdn.net/shuijiaobuzhundahulu/article/details/82257641
今日推荐