解析深度學習神經網絡權重初始化

以全連接模型爲例:輸入數據爲a[784]的數組,模型共2層神經元,第一層100個,第二層即輸出層爲10個神經元,則第一層的權重有[784,100]個,截距有[100]個,第二層有[100,10]個權重,截距[10]個,代碼用W1表示第一層權重矩陣,B1表示截距行向量,W2表示第二層權重矩陣,B2表示截距行向量,設置每次輸入101批數據,則輸入數據矩陣爲[101,784],用X表示,輸入標籤爲[10
相關文章
相關標籤/搜索