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)
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)