tensorflow基操

tf.split(A,B,C) A:表示切誰 B:表示切的方法,可以是list可以是常數 C:表示在那個維度進行切   注意TF卷積的輸入格式[batch,長,寬,深],red,gree,blue = tf.split(輸入,3,3)第二個參數是常數3表示均勻切分,第三個通道是深度,表示第三個通道也就是對顏色通道均勻切分。 tf.concat(value,dim)
相關文章
相關標籤/搜索