torch初學

參考地址:https://www.jianshu.com/p/cde4a33fa129 按照圖中的代碼敲一遍(注意view層的使用方法,以及最後計算參數數量的方式。) 參數數量的計算結果: 參數結果中的【6】、【16】、【120】、【84】、【10】全是指Bias所帶來的參數量。
相關文章
相關標籤/搜索