PyTorch Conv2d 前向传递中发生了什么？

在PyTorch中，nn.Conv2d 是一个用于二维卷积操作的模块。当你对一个输入张量（通常是一个四维的，形状为 [batch_size, channels, height, width]）调用 forward 方法（或直接使用模块作为函数调用）时，Conv2d 会执行一系列的计算来产生输出张量。

以下是 Conv2d 前向传递中发生的主要步骤：

1. 输入准备：
   • 输入张量首先被验证，以确保其形状和Conv2d模块的参数（如in_channels，即输入通道数）相匹配。
2. 卷积核（权重）与偏置：
   • Conv2d模块包含了可学习的权重（卷积核）和偏置（如果指定了的话）。
   • 这些权重和偏置在训练过程中会被更新以最小化损失函数。
3. 卷积操作：
   • 卷积操作是Conv2d的核心。它遍历输入张量的每个空间位置（高度和宽度），并将卷积核应用于该位置及其邻域。
   • 对于每个位置，输入数据的一部分（由卷积核的大小决定）与卷积核进行点积运算，然后加上偏置（如果有的话）。
   • 这个结果（称为特征映射或激活映射的一个元素）被保存到输出张量的相应位置。
4. 步长、填充