Follow me on GitHub

nn.Conv2d 参数及输入输出详解

Torch.nn.Conv2d(in_channels,out_channels,kernel_size,stride=1,padding=0,dilation=1,groups=1,bias=True)

in_channels:输入维度

out_channels:输出维度

kernel_size:卷积核大小

stride:步长大小

padding:补0

dilation:kernel间距

 

 

posted @ 2019-05-04 20:24  SiyuanChen  阅读(35217)  评论(0编辑  收藏  举报