在卷积模块中,许多参数可能会改变输入的维数。有没有什么方法可以获取卷积模块或一般模块的输出的维数?

我尝试查看SpatialConvolution中公开的方法,但是没有任何方法建议给我这些信息。

此外,输出张量似乎具有以下尺寸:

conv1 = nn.SpatialConvolution(3, 96, 5, 5, 1, 1, 2, 2)
conv1.id   = 'conv1'
print(conv1.output:size())
[torch.LongStorage of size 0]


有任何想法吗?

我意识到,除了通道数(3x32x32)外,我什至没有设置输入维数。如何设置?为什么不是强制性的?

最佳答案

输出大小取决于您的输入大小。您可以尝试如下操作:

th> require 'nngraph'
th> conv1 = nn.SpatialConvolution(3, 96, 5, 5, 1, 1, 2, 2)()
th> model = nn.gModule({conv1},{conv1})
th> x = torch.rand(3,20,20)
th> y = model:forward(x)
th> y:size()

96
20
20
[torch.LongStorage of size 3]

09-12 05:57