会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
tangjunjun
博客园
首页
新随笔
联系
订阅
管理
2023年3月2日
maxpool3d修改成maxpool2d与maxpool1d方法
摘要: 有时候遇到不支持maxpool3d的硬件或算子时候,可将其改成maxpool2d加上maxpool1d组合方式表示,经验证与maxpool3d结果完全一致,其实现细节如下: 代码: import torch class MaxPool3d_modify(torch.nn.Module): def _
阅读全文
posted @ 2023-03-02 11:23 tangjunjun
阅读(244)
评论(0)
推荐(0)
编辑
公告
https://rpc.cnblogs.com/metaweblog/tangjunjun