pytorch中F.avg_pool1d()和F.avg_pool2d()
Posted rise0111
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了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原理同上，取平均改为取最大值
以上是关于pytorch中F.avg_pool1d()和F.avg_pool2d()的主要内容，如果未能解决你的问题，请参考以下文章
CF868 F. Yet Another Minimization Problem 决策单调优化 分治
我不明白 conv1d、conv2d 的 pytorch 输入大小
pytorch 笔记 torch.nn.BatchNorm1d