torch.mul()與 torch.mm()

  torch.mul(a, b) 是矩陣a和b對應位相乘,即點乘。a和b的維度必須相等,比如a的維度是(1, 3),b的維度是(1, 3),返回的仍是(1, 3)的矩陣。 torch.mm(a, b) 是矩陣a和b矩陣相乘,即矩陣乘法。比如a的維度是(1, 3),b的維度是(3, 2),返回的就是(1, 2)的矩陣。
相關文章
相關標籤/搜索