a = torch.Tensor(np.random.normal(0,1,(2,1,1,1)))
b = torch.Tensor(np.random.normal(0,1,(2,3,4,4)))
c=a*b
c是由a和b点乘得来,先将a扩张成b的形状,数字逐位点乘。
a = torch.Tensor(np.random.normal(0,1,(2,3)))
b = torch.Tensor(np.random.normal(0,1,(3,4)))
c=torch.mm(a,b)
二维矩阵乘法,符合数学运算矩阵乘法,生成形状为(2,4)