Caffe 中convolution的計算

最後一頁沒畫,但是基本上就是Filter Matrix乘以Feature Matrix的轉置,得到輸出矩陣Cout x (H x W),就可以解釋爲輸出的三維Blob(Cout x H x W)。 Caffe裏用的是CHW的順序,有些library也會用HWC的順序(比如說CuDNN是兩個都支持的),這個在數學上其實差別不是很大,還是一樣的意思。 順便廣告一下我吐槽Caffe裏面卷積算法的鏈接:C
相關文章
相關標籤/搜索