关于李沐深度学习The size of tensor a (10) must match the size of tensor b (256) at non-singleton dimension 1
The size of tensor a (10) must match the size of tensor b (256) at non-singleton dimension 1
·
这里的话,笔者进行了代码调试,发现问题出现在W1,b1,W2,b2的代码块部分,如果你和笔者一样,开始就把W1,b1复制粘贴给W2,b2,那么请检查一下b2的 参数,为Outputs而不是Hiddens。更正后即可运行程序!
(觉得有用,麻烦点个赞再走吧)
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)