RuntimeError: mat1 dim 1 must match mat2 dim 0

这个错误呢,以我这小学三年级英语的水平,果然看不懂。但仔细一想,应该是第一个矩阵的第1维度 和 第二个矩阵的第2维度 不匹配。

既然牵涉到了矩阵变换,那应该就是神经网络定义处的错误了。

原模型

看了看没错啊,打印模型参数粗来:

也没猫饼啊。

在这里必然是网络的不匹配,那就必然找网络的茬。找到LSTM的定义函数,找到其参数解释:

哦?有个bi什么的参数,默认false,如果是true,就是bi。。。LSTM。好的,改了试试。

问题解决!

就是说,这里如果num_layer 如果不是默认的1,就将bi参数设置成True。如果是默认1,就不管即可。

可以自行试试。受益点赞。

 

 

  • 14
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值