torch.mul(a, b)
是矩阵a和b对应位相乘,a和b的维度必须相等,相乘后返回结果维度也相同torch.mm(a, b)
是矩阵a和b矩阵相乘,比如a的维度是(1, 2),b是(2, 3),返回的就是(1, 3)的矩阵
举个栗子:
import torch
a = torch.rand(1, 4)
b = torch.rand(4, 3)
c = torch.<
torch.mul(a, b)
是矩阵a和b对应位相乘,a和b的维度必须相等,相乘后返回结果维度也相同torch.mm(a, b)
是矩阵a和b矩阵相乘,比如a的维度是(1, 2),b是(2, 3),返回的就是(1, 3)的矩阵举个栗子:
import torch
a = torch.rand(1, 4)
b = torch.rand(4, 3)
c = torch.<