Tensorflow API 学习(1)-tf.concat()

函数原型 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]]

没有更多推荐了,返回首页