栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

pytorch池化层ceilmode的作用

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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,舍弃

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/991279.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号