记录 return torch.mm(x,w)+b IndexError: Dimension out of range (expected to be in range of [-1, 0...

这个错误是因为矩阵乘法中的列与另一矩阵的行数不匹配，具体的做法就是改变其中的一个变量的行列。


出现了Expected object of scalar type Double but got scalar type Float for argument #2 'mat2' in call to _th_mm，应该是数据类型不匹配，改下数据类型为double。


完美解决

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容