nn.moduleList 和Sequential由來、用法和實例 —— 寫網絡模型

對於cnn前饋神經網絡若是前饋一次寫一個forward函數會有些麻煩,在此就有兩種簡化方式,ModuleList和Sequential。其中Sequential是一個特殊的module,它包含幾個子Module,前向傳播時會將輸入一層接一層的傳遞下去。ModuleList也是一個特殊的module,能夠包含幾個子module,能夠像用list同樣使用它,但不能直接把輸入傳給ModuleList。下
相關文章
相關標籤/搜索