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