pytorch nn.Conv2d()中的padding以及輸出大小方式
更新時(shí)間:2020年01月10日 11:40:28 作者:qq_30468133
今天小編就為大家分享一篇pytorch nn.Conv2d()中的padding以及輸出大小方式,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧
我就廢話不多說了,直接上代碼吧!
conv1=nn.Conv2d(1,2,kernel_size=3,padding=1) conv2=nn.Conv2d(1,2,kernel_size=3) inputs=torch.Tensor([[[[1,2,3], [4,5,6], [7,8,9]]]]) print("input size: ",inputs.shape) outputs1=conv1(inputs) print("output1 size: ",outputs1.shape) outputs2=conv2(inputs) print("output2 size: ",outputs2.shape) 輸出: input size: torch.Size([1, 1, 3, 3]) output1 size: torch.Size([1, 2, 3, 3]) output2 size: torch.Size([1, 2, 1, 1])
padding是指卷積前進(jìn)行padding,這樣保證輸出的圖像形狀大小與輸入相同,但是通道數(shù)channel改變了。
以上這篇pytorch nn.Conv2d()中的padding以及輸出大小方式就是小編分享給大家的全部內(nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python語音合成之第三方庫gTTs/pyttsx3/speech橫評(內(nèi)附使用方法)
Python是一種非常強(qiáng)大的腳本語言,可以用來實(shí)現(xiàn)各種復(fù)雜的應(yīng)用,其中之一就是文本轉(zhuǎn)語音,即把文字轉(zhuǎn)換成聲音來發(fā)出,下面這篇文章主要給大家介紹了關(guān)于Python語音合成之第三方庫gTTs/pyttsx3/speech橫評的相關(guān)資料,文中還介紹了詳細(xì)的使用方法,需要的朋友可以參考下2023-05-05