關於torch.bmm()函數計算過程

不少框架中提供的矩陣乘法都是出於簡化計算的考慮，不少狀況下在進行計算時候都會牽扯到 batch size 這一個維度，這就使得不少矩陣的計算是三維的，Pytorch中的bmm()函數就能夠很方便的實現三維數組的乘法，而不用拆成二維數組使用for循環解決。在查資料的時候發現有些博客寫的有些小地方不太對，並且有不少提問都是關於 bmm（）函數具體是如何計算的，所以記錄。 web 文章目錄 1.torc

>>阅读原文<<