pytorch中F.avg_pool1d()和F.avg_pool2d()


F.avg_pool1d()數據是三維輸入(Batch,seq_len,embedding_size)

  • input維度: (batch_size,channels,width)channel可以看成高度

  • kenerl維度:(一維:表示width的跨度)channel和輸入的channel一致可以認為是矩陣的高度

    假設kernel_size=2,則每倆列相加求平均,stride默認和kernel_size保持一致,越界則丟棄

F.max_pool1d原理同上,取平均改為取最大值


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM