在卷积模块中,许多参数可能会改变输入的维度。有没有办法获取卷积模块输出的维度,或者一般来说,任何模块的维度?
我尝试查看SpatialConvolution中公开的方法,但没有任何方法可以提供此信息。
此外,输出张量似乎具有以下维度:
我尝试查看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
)。如何设置?为什么不是必须的?