Pytorch广播机制与matmul函数

1.pytorch的广播机制与numpy类似，当两个tensor逐元素相乘时（两个tensor的size必须相同），会遇到前后tensor的size不同，此时便需要广播机制了。 广播机制条件：其中任意一个tensor必须至少有一个维度要么相等，从两个tensor的size末尾开始比较，要么其中一个为1，要么其中一个不存在，满足其一即可。 如： 在pytorch中，*表示逐元素相乘，而@是矩阵乘积，

>>阅读原文<<