我是靠谱客的博主 俭朴戒指,最近开发中收集的这篇文章主要介绍RuntimeError: mat1 dim 1 must match mat2 dim 0,觉得挺不错的,现在分享给大家,希望可以做个参考。
概述
这个错误呢,以我这小学三年级英语的水平,果然看不懂。但仔细一想,应该是第一个矩阵的第1维度 和 第二个矩阵的第2维度 不匹配。
既然牵涉到了矩阵变换,那应该就是神经网络定义处的错误了。
原模型
看了看没错啊,打印模型参数粗来:
也没猫饼啊。
在这里必然是网络的不匹配,那就必然找网络的茬。找到LSTM的定义函数,找到其参数解释:
哦?有个bi什么的参数,默认false,如果是true,就是bi。。。LSTM。好的,改了试试。
问题解决!
就是说,这里如果num_layer 如果不是默认的1,就将bi参数设置成True。如果是默认1,就不管即可。
可以自行试试。受益点赞。
最后
以上就是俭朴戒指为你收集整理的RuntimeError: mat1 dim 1 must match mat2 dim 0的全部内容,希望文章能够帮你解决RuntimeError: mat1 dim 1 must match mat2 dim 0所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复