天天看點

pytorch學習 nn.Conv2d(nc, ndf, 4, 2, 1, bias=False)

如何了解這裡的nc和ndf呢?

首先,nc這裡是輸入的通道數,比如說圖檔的RGB三通道,當然這裡的通道數是不限的。

ndf是filter的數量,因為算的時候,就是先對每一個濾波器去計算每一個通道的,然後把每個通道的結果加和,這個和就是這個濾波器的結果,輸出的就是由ndf個濾波器算出的結果的向量。

後面三個4,2,1分别是kernel_size,stride,padding。有個公式可以算w和h,看這個文章

繼續閱讀