我想用nn.Sequential
声明性地定义我的模型。我想将卷积特征图的输出(n,c,x,y)与完全连接的层(n,c * x * y)连接时遇到问题。
如果我必须用命令定义模块,那就用
x.view(-1, c * x * y)
顺序仅接受模块,而我在TORCH.NN下找不到任何可以解决尺寸不匹配的模块。
我一直在研究Unfold
,但只能将最后两个维度(n, c, x, y)
展平为(n, c, x * y)
。
有什么方法可以通过内置模块来完成此简单任务吗? (无需编写自定义模块?)