pytorch自然語言處理之Pooling層的句子分類

Pooling作爲最簡單的層其實也可以作爲句子分類任務。Pooling有很多種,max_Pooling,avg_Pooling,min_Pooling等。常用的還是max_Pooling:取同維度的最大值。 先看看流程圖: 這裏的Linear Layer後面應該經過一個Softmax的,可是由於交叉熵cross_entropy裏隱含有Softmax,這裏我就沒有畫了。 第一步搭建網絡 這裏除了劃線
相關文章
相關標籤/搜索