
上QQ阅读APP看书,第一时间看更新
1.2.2 前向传播
定义完一个网络结构之后,后期我们会将所有数据按照batch的方式进行输入,得到对应的网络输出,这也就是所谓的前向传播,这里取少量数据样本进行举例,代码如下:

输出结果如下:

取两张图片进行网络的输入,维度情况为2×1×28×28,经过定义后进行网络输出,得到维度为10的tensor,每个位置上的数值代表成为该类别的概率值。可以看出,第二张图片在0位置的概率值最大,对应标签是数字1,而实际的标签索引为数字2,对应标签是数字3,之所以有差距是因为目前网络仍没有进行训练,只是随机初始化了结构中的权重,所以输出暂时没有考虑价值。