06torch矩阵运算

torch矩阵运算

广播机制

各种乘法运算

A*B

torch.dot(A,B)

相同位置上的乘积和

torch.dot(A,B)==torch.sum(A*B)

torch.mv(A,B)

mv=matirx vector multi

torch.mv(A,B)等同于数学上的A点乘B,然后去掉一个维度

torch.mm(A,B)

mm=matirx matrix

torch.mm(A,B)=A对B的每一层都做了一个mv操作,然后拼接

问题

copy和clone的区别

clone的程度更深。clone一定复制内存,而copy不一定

posted @ 2021-10-17 20:58  小夏的魔仙堡  阅读(311)  评论(0编辑  收藏  举报