其他分享
首页 > 其他分享> > ptorch卷积conv2d

ptorch卷积conv2d

作者:互联网

nn.Conv2d就是pytorch中定义的的一个普通的类,可以很方便的实现二维的卷积,它有五个主要的参数,

in_channels(输入的通道数)
out_channels(输出的通道数)
kernel_size(卷积核的大小)
stride(卷积核移动的步长)
padding(是否对被卷的张量做填充).
除此以外,还有
dilation:决定是否采用空洞卷积,默认为1(不采用)。这个参数的意义是从卷积核上的一个参数到另一个参数所需走过的距离,所以就默认为1.
groups:决定了是否采用分组卷积,默认为1。groups的参数决定了将输入的channels分为几组,而每组channel会被重用几次是由输入的channel数相应计算才知道,这也要求了groups需要被out_channel&in_channel整除。例如输入输出channel都为6时,设定groups=3,那么会将in_channel分为3组,每个out_channel由2个in_channel得到,每组in_channel重复使用两次。
bias:是否添加偏置,默认为true。
padding_mode:即padding的填充模式,默认采用零填充

标签:卷积,默认,channels,conv2d,groups,ptorch,channel,out
来源: https://www.cnblogs.com/jisangday1/p/16215035.html