tensorflow教程舉例中layers.GlobalAveragePooling1D()的作用

作用:是對矩陣按列進行平均池化。 比如tensorflow教程中的這個例子,中間的GlobalAveragePooling1D就是對輸入中的sequence進行平均池化。 因爲這是個分類算法,輸入是兩維的<batch, sequence>,其中batch是訓練模型的分批,sequence是單個句子的長度。 模型代碼: 模型summary: 模型簡單介紹
相關文章
相關標籤/搜索