三分鐘完全瞭解PyTorch中卷積層內部計算方法和權值初始值的來歷,讓你對模型有更深層次的理解

首先先說明第一個答案,也就是PyTorch中卷積層的計算方法,其實這點很多人可能在書上已經看過圖了,我只是用代碼復現一遍 我們把所有變量都明確,首先是輸入變量,我們設爲2 * 2的全1矩陣,如下: 然後初始化一個二維卷積層,並輸出其weight和bias,如下: 我們可以看到,weight和bias都有兩個值,這是因爲我們定義了輸出通道爲2,所以給我們分配了兩個卷積核,然後可以看到權值分別爲0.7
相關文章
相關標籤/搜索