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)的矩阵。
相关文章
相关标签/搜索