Pytorch廣播機制與matmul函數

1.pytorch的廣播機制與numpy類似，當兩個tensor逐元素相乘時（兩個tensor的size必須相同），會遇到前後tensor的size不同，此時便需要廣播機制了。 廣播機制條件：其中任意一個tensor必須至少有一個維度要麼相等，從兩個tensor的size末尾開始比較，要麼其中一個爲1，要麼其中一個不存在，滿足其一即可。 如： 在pytorch中，*表示逐元素相乘，而@是矩陣乘積，

>>阅读原文<<