nn.Conv3d


用法:

torch.nn.Conv3d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros')

Shape:

 

 

計算公式:

 

 參數:

bigotimes: 表示二維的相關系數計算 

stride: 控制相關系數的計算步長 dilation: 用於控制內核點之間的距離,詳細描述在這里 

groups: 控制輸入和輸出之間的連接: group=1,輸出是所有的輸入的卷積;

group=2,此時相當於有並排的兩個卷積層,每個卷積層計算輸入通道的一半,並且產生的輸出是輸出通道的一半,隨后將這兩個輸出連接起來。

參數kernel_sizestridepaddingdilation可以是一個int的數據 - 卷積height和width值相同,也可以是一個有三個int數據的tuple數組,tuple的第一維度表示depth的數值,tuple的第二維度表示height的數值,tuple的第三維度表示width

 

輸出后的大小計算公式:

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM