基于Pytorch的网络设计语法2

import  torch.nn as nn
import  torch.functional as F
import  torch.optim as optim
from collections  import OrderedDict

class Net2(nn.Module):# 从nn.Module  继承
    def __init__(self):# 在类的初始化函数里完成曾的构建
        super(Net2,self).__init__()
        #Sequential里面的顺序 就是前往传播的顺序
        self.conv=nn.Sequential(
            nn.Conv2d(3,32,3,1,1),
            nn.ReLU(),
            nn.MaxPool2d(2))
        self.dense= nn.Sequential(
            nn.Linear(32*3*3,128),
            nn.ReLU(),
            nn.Linear(128,10)
             )
    def forward(self,input_x):# 构建前向传播的流程
        conv_out=self.conv(input_x)
        res=conv_out.view(conv_out.size(0),-1)#拉伸处理
        out =self.dense(res)

        return  out

gsznet = Net2()
print(gsznet)
if __name__ == '__main__':
    print("XXXXXXXXXXXXXX")