Pytorch的乘法是怎樣的，針對這個問題，這篇文章詳細介紹了相對應的分析和解答，希望可以幫助更多想解決這個問題的小伙伴找到更簡單易行的方法。
a = torch.tensor([[1, 2, 3], [1, 2, 3], [1, 2, 3]]) b = torch.tensor([[1, 1, 1], [2, 2, 2], [3, 3, 3]]) c = a * b d = torch.mm(a, b) print(a) print(b) print(c) print(d)
元素對應相乘：直接用*就可以，也可以用 torch.mul(a, b)
矩陣相乘：torch.mm(a,b)此方法只適用于2維矩陣、 torch.matmul(a, b) 推薦使用此方法
一維張量的點乘：torch.dot(a,b)是對兩個為1D張量進行點積運算
關于Pytorch的乘法是怎樣的問題的解答就分享到這里了，希望以上內容可以對大家有一定的幫助，如果你還有很多疑惑沒有解開，可以關注億速云行業資訊頻道了解更多相關知識。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
