1. 程式人生 > >np.dot和torch.dot

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):只能為一維度