torch.nn.Conv2d 是 PyTorch 顶用于界说二维卷积层的类。它在卷积神经网罗（CNN）中泛泛用于料理图像数据开云(中国)Kaiyun·官方网站 - 登录入口。以下是该类的用法和参数的贵重先容：

类界说

Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros')

记着，在将样本数据传递给 Conv2d 层之前，确保其体式为 (batch_size, channels, height, width)。

参数

in_channels：输入通说念数。指定输入数据的通说念数。举例，关于RGB图像，in_channels为 3。

out_channels：输出通说念数。指定卷积层输出的特征图的通说念数。这时常决定了卷积核的数目。

kernel_size：卷积核的大小。不错是一个整数或一个元组(kH, kW)，隔离指定卷积核的高度和宽度。

stride（默许为 1）：卷积操作的步幅。不错是一个整数或一个元组 (sH, sW)，隔离指定在垂直和水时常朝上的步幅。

padding（默许为）：填充是指在愚弄卷积之前，在输入张量的周围添加零（或其他值，具体取决于填充风光）。用于猖狂输出的空间维度。不错是一个整数或一个元组 (padH, padW)，指定在输入数据的垂直和水平旯旮添加的填充像素数。

dilation（默许为 1）：彭胀率。不错是一个整数或一个元组 (dH, dW)，指定卷积核元素之间的间距。

groups（默许为 1）：分组卷积。指定卷积分组的数目。in_channels和out_channels皆必须能被groups整除。（这个参数先按默许值，当今还用不到，先岂论他。）

bias（默许为 True）：是否为每个输出通说念添加偏置项。

padding_mode（默许为‘zeros’）：用于指定填充风光。不错是 ‘zeros’、‘reflect’ 、 ‘replicate’和’circular’。

不同的填充风光会影响输入数据旯旮的料理面孔。以下是padding_mode的几个选项十分含义：

'zeros'（默许）：

– 在输入数据的旯旮填充零。这是最常用的填充风光，因为它简单且计较后果高。当不需要极端料理旯旮特征时，这是一个好的选拔。

'reflect'：

– 使用输入数据旯旮的值进行反射填充。这意味着旯旮的值会被镜像反射到填充区域。

– 这种风光有助于保留旯旮特征，因为它不会引入新的零值。

'replicate'：

– 使用输入数据旯旮的值进行复制填充。这意味着旯旮的值会被复制到填充区域。

– 这种风光也有助于保留旯旮特征，因为它不会引入新的值，仅仅复制现存的旯旮值。

circular:

– 将输入张量的左旯旮将被复制到右旯旮，上旯旮将被复制到下旯旮，依此类推，从而在卷积操作中创建一个取悦的旯旮料理，使得卷积核在界限处的操作大要研讨到输入数据的周期性特征。

提防：

• ‘zeros’（置零）：适用于一般用途，但可能会引入东说念主为界限

• ‘reflect’（反射）：能更好地保留图像的局部特征

• ‘replicate’（复制）：能平滑地扩展旯旮信息

• ‘circular’（轮回）：适用于周期性数据或特定界限的愚弄

示例：

是不是依然对这几个填充风光云里雾里、傻傻分不了了，没事儿，底下让我用一个简单的例子来诠释“torch.nn.Conv2d”中的padding_mode参数。

输入一个 3×3 的张量：

1 2 34 5 67 8 9

填充像素数 padding=1, 底下列出了每种填充风光的输出终端:

padding_mode=‘zeros’:

0 0 0 0 00 1 2 3 00 4 5 6 00 7 8 9 00 0 0 0 0

padding_mode=‘reflect’:

5 4 5 6 52 1 2 3 25 4 5 6 58 7 8 9 85 4 5 6 5

padding_mode=‘replicate’:

1 1 2 3 31 1 2 3 34 4 5 6 67 7 8 9 97 7 8 9 9

padding_mode=‘circular’:

9 7 8 9 73 1 2 3 16 4 5 6 49 7 8 9 73 1 2 3 1

代码示例

底下的例子展示了何如使用Conv2d:

import torchimport torch.nn as nn# Define a convolutional layerconv_layer = nn.Conv2d(in_channels=3, out_channels=16, kernel_size=3, stride=1, padding=1)# Example input: a batch of 2 RGB images of size 64x64input_tensor = torch.randn(2, 3, 64, 64)# Forward pass through the convolutional layeroutput_tensor = conv_layer(input_tensor)print(output_tensor.shape) # Output shape will be [2, 16, 64, 64]

创建Conv2d层后，不错将其当作神经网罗的一部分使用。以下是一个简单的例子，展示如安在神经网罗中使用Conv2d：

import torchimport torch.nn as nnimport torch.nn.functional as Fclass SimpleCNN(nn.Module):def __init__(self): super(SimpleCNN, self).__init__() self.conv1 = nn.Conv2d(3, 16, kernel_size=3, stride=1, padding=1) self.pool = nn.MaxPool2d(kernel_size=2, stride=2) self.fc1 = nn.Linear(16 * 16 * 16, 10)def forward(self, x): x = F.relu(self.conv1(x)) x = self.pool(x) x = x.view(-1, 16 * 16 * 16) x = self.fc1(x)return x# 创建网罗实例net = SimpleCNN()# 创建一个输入张量input_tensor = torch.randn(1, 3, 32, 32) # (batch_size, channels, height, width)# 前向传播output = net(input_tensor)print(output)

集会学问开云(中国)Kaiyun·官方网站 - 登录入口，津润你我。