前向传播函数
前向传播函数是什么?什么是前向传播?前向传播(Forward Propagation)就是神经网络处理输入数据、产生输出的过程。 生活类比想象一个工厂流水线: 12345678910111213原材料(输入图像) ↓工位1:提取边缘特征 ↓工位2:提取纹理特征 ↓工位3:提取高级语义 ↓质检:压缩特征 ↓包装:分类结果 ↓成品(预测类别) 这个从原材料到成品的整个过程,就是前向传播。 在代码中的体现1234567891011121314151617181920def forward(self, x): """ 前向传播函数 参数: x: 输入张量,形状为 (batch_size, 3, 64, 64) 返回: 输出张量,形状为 (batch_size, num_classes) """ # 第一层卷积特征提取 x = self.conv1(x) # 第二层卷积特征提取 x = self.conv2(x)...