關於pytorch中nn.conv中的參數

關於pytorch中nn.conv中的參數 之前一直是用的tensor flow +keras,最近剛轉爲pytorch,發現torch中的卷積參數和tensorflow中的有點不同,特來記錄下,以下是官網中的解釋: 和tensorflow不同,in_channels和out_channels分別是輸入和輸出的通道數,stride步長以及padding填充和keras是一樣的,groups 這個參
相關文章
相關標籤/搜索