神经网络学习之 concat函数

  在Dense NetU Net中,常用concat函数

  在 dense net 中,concat操作如下:

        out = tf.concat((input1, input2, input3), 3)

    u net 网络(如下),蓝色框选处,也是concat操作

神经网络学习之 concat函数_第1张图片

   在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]]

 


你可能感兴趣的:(神经网络学习之 concat函数)