Torch cv 用法

nn.Conv2d

conv_layer = nn.Conv2d(
    in_channels=3,       # 输入通道数
    out_channels=16,     # 输出通道数
    kernel_size=3,       # 卷积核大小
    stride=1,            # 步长
    padding=1,           # 填充
    dilation=1,          # 膨胀
    groups=1,            # 组数
    bias=True            # 是否使用偏置
)