用于重塑的PyTorch模块

时间:2019-03-03 16:53:03

标签: python pytorch

我想用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)

有什么方法可以通过内置模块来完成此简单任务吗? (无需编写自定义模块?)

0 个答案:

没有答案