在PyTorch中,我想创建一个神经元未完全连接到输出层的隐藏层。我尝试串联两个线性层的输出,但遇到以下错误:
RuntimeError:大小不匹配,m1:[2 x 2],m2:[4 x 4]
我当前的代码:
Module
我不明白为什么尺寸不匹配?有没有其他方法可以在pytorch中实现非完全连接的层?
答案 0 :(得分:0)
事实证明,使用串联功能是一个简单的理解问题。
改变中
x = torch.cat((xLeft, xRight))
至
x = torch.cat((xLeft, xRight), dim=1)
做到了。
谢谢@dennlinger