pytorch学习——Conv2d卷积

Conv2d()

Ctrl + B 进入到torch.nn.Conv2d()，可以发现Conv2d()是一个类，如下

class Conv2d(_ConvNd):
    def __init__(self, in_channels, out_channels, kernel_size, stride=1,
                     padding=0, dilation=1, groups=1,
                     bias=True, padding_mode='zeros'):
        pass

参数：
in_channels：输入图片的通道数量(int)
out_channels：卷积后图片的通道数量(int)
kernel_size：卷积核大小(int or tuple))
stride: 卷积步长，默认为1(int or tuple, optional))
padding: 在输入的边界进行填充，默认填充为零(int or tuple, optional)

Args:
in_channels (int): Number of channels in the input image
out_channels (int): Number of channels produced by the convolution
kernel_size (int or tuple): Size of the convolving kernel
stride (int or tuple, optional): Stride of the convolution. Default: 1
padding (int or tuple, optional): Zero-padding added to both sides of the input. Default: 0
padding_mode (string, optional): 'zeros', 'reflect', 'replicate' or 'circular'. Default: 'zeros'
dilation (int or tuple, optional): Spacing between kernel elements. Default: 1
groups (int, optional): Number of blocked connections from input channels to output channels. Default: 1
bias (bool, optional): If True, adds a learnable bias to the output. Default: ``True`

nn.BatchNorm2d

nn.Dropout2d

MaxPool2d
nn.AvgPool2d
nn.Sequential()
nn.Linear
nn.ModuleList
nn.AdaptiveAvgPool2d

后续更新…