深度學習(九)——Tensor基本運算

+:add或者+ -:sub或者- *:mul或者* /:div或者/——一個**/符號代表除法,兩個//**代表整除 矩陣相乘 *:是對應位置相乘 torch.mm只適用2d,不推薦 [email protected]:矩陣相乘,@是matmul的重載 圖中w爲(512,784),a(4,784)@w(784,512)=(4,512),爲什麼w的512放第一位呢:Pytorch習慣把輸出放第一位
相關文章
相關標籤/搜索