pytorch笔记:torch.nn.MaxPool2d

1 基本用法

class torch.nn.MaxPool2d(
    kernel_size, 
    stride=None, 
    padding=0, 
    dilation=1, 
    return_indices=False, 
    ceil_mode=False)

2 参数详解

kernel_size

        最大池化窗口的大小,类型为int 或者元组,

        当最大池化窗口是方形的时候,只需要一个整数边长即可;最大池化窗口不是方形时,要输入一个元组表示 高和宽。【必选】

        每次将输入池化操作的kernel_size大小的内容进行最大池化操作得到一个结果

stride

最大池化窗口每次滑动的步长为多少,默认是 kernel_size【可选】

以kernel_size为1为例,会有:

这一轮考虑的是a[i][j],下一轮考虑的就是a[i][j+stride]

padding

设置在所有边界增加 值为 0 的边距的大小(也就是在feature map 外围增加几圈 0 )

例如当 padding =1 的时候,如果原来大小为 3 × 3 ,那么之后的大小为 5 × 5 。即在外围加了一圈 0 。【可选】

padding的作用是为了维护输出的维度

dilation

控制最大池化窗口之间的间距【可选】

如果我们设置的dilation=0的话,效果如图:(蓝色为输入,绿色为输出,最大池化窗口为3 × 3)

如果设置的是dilation=1,那么效果如图:(蓝色为输入,绿色为输出,最大池化窗口卷积核仍为 3 × 3 。)
但是这里最大池化窗口点与输入之间距离为1的值相乘来得到输出。

ceil_mode如果等于True,计算输出信号大小的时候,会使用向上取整,代替默认的向下取整的操作

3 输入输出形状

假设输入形状为:,输出形状为

(batchsize, channel, height, width)

那么,根据MaxPool2d的参数,我们有:

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

UQI-LIUWJ

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值