feedforward_neural_network

CPU版本

前面简单介绍了用PyTorch进行逻辑回归和线性回归，在逻辑回归中使用一个线性层对mnist数据集进行了分类，其实就是一个简单的神经网络，模子都是一样的，在这一节中我们使用两个线性层来看看如何构造。前面简单