深度学习(九)——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习惯把输出放第一位
相关文章
相关标签/搜索