使用pytorch框架實現多層感知機

和實現softmax回歸唯一的不同在于我們多加了一個全連接層作為隱藏層。它的隱藏單元個數為256，并使用ReLU函數作為激活函數。

#模型的核心代碼為:nn.Linear(num_inputs, num_hiddens),nn.ReLU(),nn.Linear(num_hiddens, num_outputs),

總結

以上是生活随笔為你收集整理的深度学习pytorch--多层感知机(三)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。