pytorch池化层ceilmode的作用

rehsape函数有一个比较特殊的-1,这个在许多代码上也是使用的比较频繁
关于-1这部分理解参考这篇文章:
1分钟理解pytorch的reshape函数中-1表示的意义
-1的值为总元素个数除以(其他维度值相乘)
比如原始tensor维度为(2,2,3),reshape为(-1,3,4),-1的值就为12/(3×4)=1,即reshape后的tensor的shape为(1,3,4)

* 当ceil_mode = True时:

             0   0 | 0   0 | 0   ×
             0  1  | 0   1 | 1   ×
           -----------------------
             2   2 | 2   2 | 2   ×
             3   3 | 3   3 | 3   ×
           -----------------------
             4   4 | 4   4 | 4   ×
             × ×   | ×   × | ×   ×

    输出:输出3*3维
        [1 1 1
         3 3 3
         4 4 4]
    即:数据不足以构成2*2,仍然对剩余数据进行计算
* 当ceil_mode = False时:输出2*2维
       0   0 | 0    0 
       1   1 | 1    1 
       --------------
       2   2 |  2   2 
       3   3 |  3   3 
    输出:
        [1   1
         3  3]
    即:数据不足以构成2*2,舍弃

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

同学yes

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

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

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

打赏作者

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

抵扣说明:

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

余额充值