对于全连接层,是否可以在 caffe 中输出类似 512x3x3 的形状
Is it possible to have an output shape like 512x3x3 in caffe for a Fully connected layer
我一直在阅读 this paper,我看到他们使用了一个尺寸为 1x1x1024 的全连接层,然后在下一步中他们突然有了 512x3x3x3。这怎么可能?你会如何在咖啡中做到这一点?
我找到了论文的代码。
他们所做的是像这样在全连接层之后重塑层:
voxel_dec:add(nn.Linear(opt.nz, 3*3*3*512))
voxel_dec:add(nn.ReLU())
voxel_dec:add(nn.Reshape(512, 3, 3, 3))
我一直在阅读 this paper,我看到他们使用了一个尺寸为 1x1x1024 的全连接层,然后在下一步中他们突然有了 512x3x3x3。这怎么可能?你会如何在咖啡中做到这一点?
我找到了论文的代码。 他们所做的是像这样在全连接层之后重塑层:
voxel_dec:add(nn.Linear(opt.nz, 3*3*3*512))
voxel_dec:add(nn.ReLU())
voxel_dec:add(nn.Reshape(512, 3, 3, 3))