零基礎入門深度學習-dive in to pytorch 的程式碼報錯記錄

maoni99999發表於2020-10-02

李沐,亞馬遜AI專家聯手打造了動手學習深度學習一書,該書的原版程式碼是用MXnet框架寫的,國內北大有學生翻譯成了pytorch版本(太強了),便於學習,但其中因為pytorch的版本更新,有幾處的程式碼編譯通過不了

 

報錯記錄:來自知乎 逆風的香 的整理

(1)第3節

https://zhuanlan.zhihu.com/p/90590957

錯誤程式碼:RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #3 'mat2' in call to _th_addmm_out

加上

    X = torch.tensor(X, dtype=torch.float32)

發現。。還是不太行,是pytoch 的版本太新了嗎Hhhh,錯誤程式碼:

RuntimeError: matrices expected, got 1D, 2D tensors at ..\aten\src\TH/generic/THTensorMath.cpp:36

錯誤源來自

torch.mm(X, w) + b

意思是,mm()目前只能進行一維點乘運算?

有大佬解答嗎

相關文章