pytorch实现特殊的Module--Sqeuential三种写法-创新互联

我就废话不多说了,直接上代码吧!

集宁ssl适用于网站、小程序/APP、API接口等需要进行数据传输应用场景,ssl证书未来市场广阔!成为创新互联的ssl证书销售渠道,可以享受市场价格4-6折优惠!如果有意向欢迎电话联系或者加微信:18982081108(备注:SSL证书合作)期待与您的合作!
# -*- coding: utf-8 -*-
#@Time  :2019/7/1 13:34
#@Author :XiaoMa
 
import torch as t
from torch import nn
#Sequential的三种写法
net1=nn.Sequential()
net1.add_module('conv',nn.Conv2d(3,3,3))  #Conv2D(输入通道数,输出通道数,卷积核大小)
net1.add_module('batchnorm',nn.BatchNorm2d(3))  #BatchNorm2d(特征数)
net1.add_module('activation_layer',nn.ReLU())
 
net2=nn.Sequential(nn.Conv2d(3,3,3),
          nn.BatchNorm2d(3),
          nn.ReLU()
          )
 
from collections import OrderedDict
net3=nn.Sequential(OrderedDict([
  ('conv1',nn.Conv2d(3,3,3)),
  ('bh2',nn.BatchNorm2d(3)),
  ('al',nn.ReLU())
]))
 
print('net1',net1)
print('net2',net2)
print('net3',net3)
 
#可根据名字或序号取出子module
print(net1.conv,net2[0],net3.conv1)

另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。


当前标题:pytorch实现特殊的Module--Sqeuential三种写法-创新互联
网站网址:http://pwwzsj.com/article/cchgip.html