使用pytorch构建一个简单的神经网络

网络的搭建 首先，咱们创造一个继承于类 nn.Module 下的类 Net，并在初始化时就定义网络的结构。咱们构建出两层卷积层和三个全链接层做为例子web def __init__(self): super(Net, self).__init__() # 1 input image channel, 6 output channels, 3x3 square con

>>阅读原文<<