+:add或者+
-:sub或者-
*
:mul或者*
/:div或者/——一个**/符号代表除法,两个//**代表整除
矩阵相乘
*
:是对应位置相乘
torch.mm只适用2d,不推荐
torch.matmul=@:矩阵相乘,@是matmul的重载
图中w为(512,784),a
(4,784)@w
(784,512)=(4,512),为什么w的512放第一位呢:Pytorch习惯把输出放第一位,t()
:转置,适合2d,高维用transpose
四维*四维:前两位不变,矩阵乘后两位,前两位如果不一致就适用broadcast
深度学习(九)——Tensor基本运算
最新推荐文章于 2023-02-26 00:31:25 发布