5.1块和层
我们一直在通过net(X)
调用我们的模型来获得模型的输出。 这实际上是net.__call__(X)
的简写。 这个前向传播函数非常简单: 它将列表中的每个块连接在一起,将每个块的输出作为下一个块的输入。
import torch
from torch import nn
from torch.nn import functional as F
net = nn.Sequential(nn.Linear(20, 256), nn.ReLU(), nn.Linear(256, 10))
X = torch.rand(2, 20)
net(X)
块的一个主要优点是它的多功能性。 我们可以子类化块以创建层(如全连接层的类)、 整个模型(如上面的MLP
类)或具有中等复杂度的各种组件。
Sequential
的设计则是为了把其他模块串起来。
5.1练习
1如果将MySequential
中存储块的方式更改为Python列表,会出现什么样的问题?
2实现一个块,它以两个块为参数,例如net1
和net2
,并返回前向传播中两个网络的串联输出。这也被称为平行块。
3假设我们想要连接同一网络的多个实例。实现一个函数,该函数生成同一个块的多个实例,并在此基础上构建更大的网络。
标签:20,nn,AI,torch,学习,Sequential,深度,import,net From: https://www.cnblogs.com/CLGYPYJ/p/17557791.html