我们可以通过torch.nn package构建神经网络. 现在我们已经了解了autograd,nn基于autograd来定义模型并对他们有所区分. 一个 nn.Module模块由如下部分构成:若干层,以及返回output的forward(input)方法. 例如,这张图描述了进行数字图像分类的神经网络: 这是一个简单的前馈( feed-forward)网络,读入input内容,每层接受前一级的输入,并输出到下一级,直到给出outpu结果. 一个经典神经网络的训练程序如下: 1.定义具有可学习参…