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):只能为一维度