会员
众包
新闻
博问
闪存
赞助商
HarmonyOS
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
会员中心
简洁模式
...
退出登录
注册
登录
博客园
首页
新随笔
联系
订阅
管理
2022年11月3日
卷积层权重的形状
摘要: conv = torch.nn.Conv2d(180, 2, (3, 3), stride=1, padding=1) print(conv.weight.data.shape) #卷积核权重是一个四维张量,第一个维度表示卷积核个数,第二个维度表示每个卷积核的层数,第三和第四个维度表示卷积核的宽 输
阅读全文
posted @ 2022-11-03 15:58 沈香茶
阅读(185)
评论(0)
推荐(0)