PyTorch:创建非完全连接层/连接隐藏层的输出

时间:2018-10-25 16:52:42

标签: python neural-network pytorch

在PyTorch中,我想创建一个神经元未完全连接到输出层的隐藏层。我尝试串联两个线性层的输出,但遇到以下错误:

  

RuntimeError:大小不匹配,m1:[2 x 2],m2:[4 x 4]

我当前的代码:

Module

我不明白为什么尺寸不匹配?有没有其他方法可以在pytorch中实现非完全连接的层?

1 个答案:

答案 0 :(得分:0)

事实证明,使用串联功能是一个简单的理解问题。 改变中     x = torch.cat((xLeft, xRight)) 至     x = torch.cat((xLeft, xRight), dim=1) 做到了。 谢谢@dennlinger