基于Pytorch的网络设计语法3

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

class Net3(nn.Module):# 从nn.Module  继承
    def __init__(self):# 在类的初始化函数里完成曾的构建
        super(Net3,self).__init__()
        #Sequential里面的顺序 就是前往传播的顺序
        self.conv=nn.Sequential()
        self.conv.add_module("conv1",nn.Conv2d(3,32,3,1,1))
        self.conv.add_module("relu1", nn.ReLU())
        self.conv.add_module("pool1", nn.MaxPool2d(2))
        self.dense = nn.Sequential()
        self.dense.add_module("dense1", nn.Linear(32*3*3,128))
        self.dense.add_module("relu2", nn.ReLU())
        self.dense.add_module("dense2",  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 = Net3()
print(gsznet)
if __name__ == '__main__':
    print("XXXXXXXXXXXXXX")