[Pytorch] 3D卷积尺寸计算
Posted geoli
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了[Pytorch] 3D卷积尺寸计算相关的知识，希望对你有一定的参考价值。
公式来自官方文档，戳->（Conv3d — PyTorch master documentation）
本文仅作记录，顺便练习Latex语法
(D_{out}=frac{D_{in}+2 imes padding[0]-dilation[0] imes(kernel\_size[0]-1)-1}{stride[0]}-1)
(H_{out}=frac{H_{in}+2 imes padding[1]-dilation[1] imes(kernel\_size[1]-1)-1}{stride[1]}-1)
(W_{out}=frac{W_{in}+2 imes padding[2]-dilation[2] imes(kernel\_size[2]-1)-1}{stride[2]}-1)
如果默认dilation=1的话：
(D_{out}=frac{D_{in}+2 imes padding[0]-(kernel\_size[0]-1)-1}{stride[0]}-1)
(H_{out}=frac{H_{in}+2 imes padding[1]-(kernel\_size[1]-1)-1}{stride[1]}-1)
(W_{out}=frac{W_{in}+2 imes padding[2]-(kernel\_size[2]-1)-1}{stride[2]}-1)
如果默认dilation=1, stride=1的话：
(D_{out}=D_{in}+2 imes padding[0]-(kernel\_size[0]-1)-2)
(H_{out}=H_{in}+2 imes padding[1]-(kernel\_size[1]-1)-2)
(W_{out}=W_{in}+2 imes padding[2]-(kernel\_size[2]-1)-2)
以上是关于[Pytorch] 3D卷积尺寸计算的主要内容，如果未能解决你的问题，请参考以下文章
PyTorch快速搭建多个具有相同卷积核尺寸不同卷积核数量的卷积层