PyTorch基础——二维卷积时的卷积核以及输入tensor的shape

3通道100行200列的一幅图像img为例,期望输出16通道卷积特征,卷积核为3行3列,则
(1)函数调用:
          conv = torch.nn.Conv2d(in_channels=3,  out_channels=16,  kernel_size=3)
          out    = conv(img)
其中,nn.Conv2d函数中
         第一个参数表示输入图像的通道数
         第二个参数表示输出图像的通道数
         第三个参数表示卷积核尺寸,行列相等时可直接缩写为一个数
(2)卷积核的shape:   out_channels x in_channels x kernel_size1 x kernel_size2 ==> 16 x 3 x 3 x3
         输入img的shape: sample_count x in_channels x height x width                      ==>   1 x 3 x 100 x 200
conv中已包含初始化的卷积核,无需手动输入

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值