CONV2D
官网:https://pytorch.org/docs/stable/generated/torch.nn.Conv2d.html
torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros', device=None, dtype=None)
- in_channels - 输入通道数
- out_channels - 输出通道数
- kernel_size – 卷子核的大小
- stride – 步长
- padding – 指定在输入数据的四个边上添加的填充(padding)数量
- padding_mode –
'zeros'
,'reflect'
,'replicate'
or'circular'
. Default:'zeros'
padding填充的方式 - dilation – 用于指定卷积核元素之间的间隔
- groups – 用于指定卷积操作中的分组卷积
- bias - 指定是否在卷积操作中添加一个可学习的偏置(bias)