nn.moduleList 和Sequential由来、用法和实例 —— 写网络模型

对于cnn前馈神经网络若是前馈一次写一个forward函数会有些麻烦,在此就有两种简化方式,ModuleList和Sequential。其中Sequential是一个特殊的module,它包含几个子Module,前向传播时会将输入一层接一层的传递下去。ModuleList也是一个特殊的module,能够包含几个子module,能够像用list同样使用它,但不能直接把输入传给ModuleList。下
相关文章
相关标签/搜索