Pytorch 中nn.Module类(nn.sequential)

nn.Module类是所有网络结构层次的父类。

当你要实现一个自己的层的时候,必须要继承这个类。

作用:

1.nn.sequential()用作容器Container

Sequential 允许我们构建序列化的模块。就把Sequential当作list来看。nn.sequential(), 一个有序的容器,神经网络模块将按照在传入构造器的顺序依次被添加到计算图中执行。

Sequential 允许我们构建序列化的模块,也就是说用了Sequential的好处是我们可以通过数字访问第几层,可以通过parameters、weights等参数显示网络的参数和权重

使用和不使用nn.Sequential()的对比

不使用

class Net(nn.Module):

def __init__(self):
    super(Net, self).__init__()

    # xw+b
    self.fc1 = nn.Linear(28*28, 256)
    self.fc2 = nn.Linear(256, 64)
    self.fc3 = nn.Linear(64, 10)

def forward(self, x):
    # x: [b, 1, 28, 28]
    # h1 = relu(xw1+b1)
    x = F.relu(self.fc1(x))
    # h2 = relu(h1w2+b2)
    x = F.relu(self.fc2(x))
    # h3 = h2w3+b3
    x = self.fc3(x)

    return x

使用

class Net(nn.Module):

def __init__(self):
    super(Net, self).__init__()

    self.fc = nn.Sequential(
    	nn.Linear(28*28, 256)
    	nn.Relu()
        nn.Linear(256, 64)
        nn.Relu()
        nn.Linear(64, 10)
    	)

def forward(self, x):
    # x: [b, 1, 28, 28]
    # h1 = relu(xw1+b1)
    x = self.fc(x)

    return x

参考https://blog.csdn.net/hxxjxw/article/details/106231242

相关推荐
封步宇AIGC13 分钟前
量化交易系统开发-实时行情自动化交易-Okex K线数据
人工智能·python·机器学习·数据挖掘
封步宇AIGC15 分钟前
量化交易系统开发-实时行情自动化交易-Okex交易数据
人工智能·python·机器学习·数据挖掘
z千鑫17 分钟前
【人工智能】利用大语言模型(LLM)实现机器学习模型选择与实验的自动化
人工智能·gpt·机器学习·语言模型·自然语言处理·自动化·codemoss
小爬虫程序猿17 分钟前
如何利用Python解析API返回的数据结构?
数据结构·数据库·python
shelly聊AI19 分钟前
AI赋能财务管理,AI技术助力企业自动化处理财务数据
人工智能·财务管理
波点兔20 分钟前
【部署glm4】属性找不到、参数错误问题解决(思路:修改模型包版本)
人工智能·python·机器学习·本地部署大模型·chatglm4
佚明zj1 小时前
全卷积和全连接
人工智能·深度学习
一点媛艺3 小时前
Kotlin函数由易到难
开发语言·python·kotlin
qzhqbb4 小时前
基于统计方法的语言模型
人工智能·语言模型·easyui
冷眼看人间恩怨4 小时前
【话题讨论】AI大模型重塑软件开发:定义、应用、优势与挑战
人工智能·ai编程·软件开发