tensorflow隨筆——concat(), stack(), unstack()

tf.concat用於將多個張量在某維度合併起來,類似於numpy.concatenate。第一個參數爲待合併的多個張量列表,第二個參數需指定哪個維度上做合併。待合併的張量shape需要完全相同,並且產生的張量的階數不會發生變化。 tf.stack(values, axis=0, name='stack')以指定的軸axis,將一個維度爲R的張量數組轉變成一個維度爲R+1的張量。即會在新的張量階上
相關文章
相關標籤/搜索