nn.Sequential作用
# 介绍
nn.Sequential是PyTorch中的一个序列容器,用于搭建神经网络的模块被按照被传入构造器的顺序添加到nn.Sequential()容器中。 除此之外,一个包含神经网络模块的OrderedDict也可以被传入nn.Sequential()容器中。 利用nn.Sequential()搭建好模型架构,模型前向传播时调用forward()方法,模型接收的输入首先被传入nn.Sequential()包含的第一个网络模块中。 然后,第一个网络模块的输出传入第二个网络模块作为输入,按照顺序依次计算并传播,直到nn.Sequential()里的最后一个模块输出结果。
# 作用
将整个容器视为单个模块,相当于把多个模块封装成一个模块,forward()方法接受输入之后,nn.Sequential()按照内部模块的顺序自动依次计算并输出结果
编辑 (opens new window)
上次更新: 2024/05/30, 07:49:34