np.dot和torch.dot

numpy.dot(a, b, out=None) 
    计算两个数组的乘积。对于二维数组来说,dot()计算的结果就相当于矩阵乘法。对于一维数组,它计算的是两个向量的点积。 对于N维数组,它是a的最后一维和b的倒数第二维和的积对于N维数组,它是a的最后一维和b的倒数第二维和的积:dot(a, b)[i,j,k,m] = sum(a[i,j,:] * b[k,:,m])。

 

 

如果a和b都是1-D阵列,则它是它们的内积(没有复共轭)。例如:

 

如果a和b都是2-D阵列,则它是矩阵乘法。

 

如果a或b是0-D(标量),则相当于乘法并使用numpy.multiply(a,b)或a * b是首选。

 

 

如果a是N-D阵列而b是1-D阵列,则它是a和b的最后一个轴上的和积。

 

 

如果a是N-D数组且b是M-D数组(其中M> = 2),则它是a的最后一个轴和b的倒数第二个轴的和积:

torch.dot(a,b):只能为一维度

猜你喜欢

转载自blog.csdn.net/weixin_40957334/article/details/82822903
今日推荐