火炬：如何在3D中重塑4D张量

Question

我需要将尺寸为x x x x x d的4D张量重新整形为3D列表，a * b x c x d。我需要它们在每个卷积层之后可视化过滤器，使用以下代码：

 for k,v in pairs(model:findModules('cudnn.SpatialConvolution')) do
   filters = v.weight
   image.save("PathWhereToSave", image.scale(image.toDisplayTensor{input=filters}, 224, 224))
 end   

不幸的是，由于张量设置的方式，我收到了这个错误：

  

打包必须是HxW或KxHxW或Kx3xHxW张量，或张量列表

您是否有一些建议或更好的方法来做我需要做的事情？ 提前致谢