pytorch學習筆記

  關於python中帶下劃線的變量和函數 Conv2d、Conv3d區別 pytorch官方文檔說明: Conv2d一般是二維平面圖像處理。默認處理的輸入輸出的shape是四維:batch size、通道數(RGB-3通道或者灰度-1通道)、圖像高度、圖像寬度。 用法是:nn.Conv2d(1,64,3)       #1表示輸入的通道數,64表示輸出的通道數,3表示卷積核爲3x3 Conv3d
相關文章
相關標籤/搜索