Python学习-乘法操作:点乘和叉乘

目录

简介

点乘(对应元素相乘)

 参考


简介

最近有点分不清点乘和叉乘。已知torch.Tensor有4种常见的乘法:*, torch.mul, torch.mm, torch.matmul. 本文抛砖引玉,简单叙述一下这4种乘法的区别,具体使用还是要参照官方文档。

点乘

  • a与b做*乘法,原则是如果a与b的size不同,则以某种方式将a或b进行复制,使得复制后的a和b的size相同,然后再将a和b做element-wise的乘法。(但是矩阵的话维度必须完全相同,标量

你可能感兴趣的:(pyg)