网络容器------Pytorch

一、容器(在Net网络类的__init__函数中使用)
    1、nn.Sequetial:按顺序包装多个网络层
        class Net(nn.module):
            def __init__(self,classes)
                super(Net,self).__init__()
                self.network = nn.Sequetial(
                    nn.Conv2d(3,6,5),
                    nn.ReLU())
                self.network.add_module('MaxPool',nn.MaxPool2d(kernel_size=2,stride=2))
                self.network.add_module(nn.MaxPool2d(kernel_size=2,stride=2))
            def forward(self,x):
                x = self.network(x)
                x = x.view(x.size()[0],-1)
            
    2、nn.ModuleList:像python的list一样包装多个网络层
        2.1 extend() # 拼接两个ModuleList
        2.2 insert() # 在指定ModuleList位置插入网络层
        class Net(nn.module):
            def __init__(self)
                super(Net,self).__init__()
                self.linears = nn.ModuleList(
                    nn.Linear(10,10) for i in range(3)])
                self.linears.append(nn.Linear(10,1))
            def forward(self,x):
                for linear in self.linears:
                    x = linear(x)
                return x
                
    3、nn.moduleDict:像python的dict一样包装多个网络层
        # clear() 清空ModuleDict
        # items() 返回可迭代的键值对
        # keys() 返回字典的键
        # values() 返回字典的值
        # pop() 返回一对键值,并从字典中删除





其他
    1、torch.nn.Parameter:张量子类,表示可学习参数,如weight,bias
    2、torch.nn.Module:所有网络基类,管理网络属性
    3、torch.nn.functional:函数具体实现,如卷积、池化、激活函数等
    4、torch.nn.init:参数初始化方法

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值