使用pytorch構建一個簡單的神經網絡

網絡的搭建 首先，咱們創造一個繼承於類 nn.Module 下的類 Net，並在初始化時就定義網絡的結構。咱們構建出兩層卷積層和三個全鏈接層做爲例子web def __init__(self): super(Net, self).__init__() # 1 input image channel, 6 output channels, 3x3 square con

>>阅读原文<<