【发布时间】:2020-03-26 09:27:47
【问题描述】:
我想知道Torch(即nn.SpatialConvolution)中的卷积层和Pytorch(即torch.nn.Conv2d)中的卷积层之间的区别
在Torch的文档中,我找到了SpatialConvolution的输出形状
它说“如果输入图像是 3D 张量 nInputPlane x height x width,则输出图像大小将为 nOutputPlane x oheight x owidth 其中
owidth = floor((width + 2*padW - kW) / dW + 1)
oheight = floor((height + 2*padH - kH) / dH + 1)
"
这与 Pytorch Docs 中的 torch.nn.Conv2d 不同。
这是否意味着它们是不同的操作?
【问题讨论】:
标签: machine-learning conv-neural-network pytorch torch