在Dense Net和U Net中,常用concat函数
在 dense net 中,concat操作如下:
out = tf.concat((input1, input2, input3), 3)
u net 网络(如下),蓝色框选处,也是concat操作
在U Net中的代码。
out = concatenate([ UpSampling2D( size=(2,2), conv2d_8), conv2d_2], axis=3 )
函数使用说明:
tf.concat(concat_dim,values,name='concat')
参数:
concat_dim:值范围是[0,rank(values)-1]。等于0,表示沿着第一维连接tensor;等于1,表示沿着第二维度连接tensor,……
values:tensor值列表
例子:
x=tf.constant([[1,2,3],[4,5,6]])
y=tf.constant([[7,8,9,10],[11,12,13,14]])
tf.concat(0,[x,y])=>[[1,2,3],[4,5,6],[7,8,9,10],[11,12,13,14]]
tf.concat(1,[x,y])=>[[1,2,3,7,8,9,10],[4,5,6,11,12,13,14]]