快捷導(dǎo)航

python中使用numpy包的向量矩陣相乘np.dot和np.matmul實(shí)現(xiàn)

更新時(shí)間：2023年02月15日 09:38:08 作者：ViviranZ

本文主要介紹了python中使用numpy包的向量矩陣相乘np.dot和np.matmul實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

1.向量和矩陣

在numpy中，一重方括號(hào)表示的是向量vector，vector沒(méi)有行列的概念。二重方括號(hào)表示矩陣matrix，有行列。

代碼顯示如下：

import numpy as np
a=np.array([1,2,3])
a.shape
#(3,)
b=np.array([[1,2,3],[3,4,5]])
b.shape
#(2, 3)
c=np.array([[1],[2],[3]])
c.shape
#(3, 1)

即使[1,2,3]、[[1,2,3]]看起來(lái)內(nèi)容一樣使用過(guò)程中也會(huì)有完全不一樣的變化。下面以向量乘法為例解釋。

2.向量和向量乘法

1.* 對(duì)應(yīng)對(duì)應(yīng)位置相乘

普通的*：在numpy里表示普通的對(duì)應(yīng)位置相乘，注意相乘的兩個(gè)向量、矩陣要保證維數(shù)相同

a1=np.array([1,2,3])
a2=np.array([1,2,3])
a1*a2
#array([1, 4, 9])
 
b1=np.array([[1,2,3]])
b2=np.array([[1,2,3]])
b1*b2
#array([[1, 4, 9]])
 
b1=np.array([[1,2,3],[3,4,5]])
b2=np.array([[1,2,3],[3,4,5]])
b1*b2
# array([[ 1,  4,  9],
#        [ 9, 16, 25]])

2.廣播機(jī)制

如果單純出現(xiàn)維數(shù)對(duì)不上，python會(huì)報(bào)error

b1=np.array([[1,2]])
b2=np.array([[1,2,3]])
b1*b2
#operands could not be broadcast together with shapes (1,2) (1,3)

但是，還有一種情況會(huì)出現(xiàn)乘出來(lái)一個(gè)好大的矩陣，這個(gè)情況常出現(xiàn)在無(wú)意之中把行、列的數(shù)字搞反的情況下。被稱為廣播機(jī)制，需要兩個(gè)乘子都有一個(gè)維數(shù)是1，如果是對(duì)不上且不為1就會(huì)報(bào)錯(cuò)

Numpy中的廣播機(jī)制，你確定正確理解了嗎？

在普通的對(duì)應(yīng)位置相乘，會(huì)出現(xiàn)

a1=np.array([1,2,3])
a3=np.array([[1],[2],[3]])
a1*a3#broadcast together
# array([[1, 2, 3],
#        [2, 4, 6],
#        [3, 6, 9]])

倒過(guò)來(lái)也會(huì)出現(xiàn)

a1=np.array([1,2,3])
a3=np.array([[1],[2],[3]])
a3*a1#broadcast together
# array([[1, 2, 3],
#        [2, 4, 6],
#        [3, 6, 9]])

3.向量點(diǎn)乘np.dot

必須要(行向量，列向量）形式的輸入

a1=np.array([1,2,3])
a3=np.array([[1],[2],[3]])
np.dot(a3,a1)
#array([14])
#ValueError: shapes (3,1) and (3,) not aligned: 1 (dim 1) != 3 (dim 0)

都是行向量，不行

b1=np.array([[1,2,3]])
b2=np.array([[1,2,3]])
np.dot(b1,b2) 
#shapes (1,3) and (1,3) not aligned: 3 (dim 1) != 1 (dim 0)

都是列向量，觸發(fā)廣播機(jī)制

a1=np.array([[1,2,3]])
a3=np.array([[1],[2],[3]])
np.dot(a3,a1)
# array([[1, 2, 3],
#        [2, 4, 6],
#        [3, 6, 9]])

3.矩陣和向量乘法

1.對(duì)應(yīng)位置相乘

如果單純采用*的方式進(jìn)行矩陣和向量乘法，那就是廣播機(jī)制

矩陣+向量

A1=np.array([[1,2,3],[2,3,4]])
b1=np.array([1,2,3])
A1*b1 #broadcast together
# array([[ 1,  4,  9],
#        [ 2,  6, 12]])

對(duì)應(yīng)的向量如果是矩陣形式，結(jié)果相同

A2=np.array([[1,2,3],[2,3,4]])
b2=np.array([[1,2,3]])
A2*b2 #broadcast together
# array([[ 1,  4,  9],
#        [ 2,  6, 12]])

相似的，如果維數(shù)對(duì)不上，不會(huì)觸發(fā)廣播機(jī)制

A3=np.array([[1,2,3],[2,3,4]])
b3=np.array([[1],[2],[3]])
A3*b3 #operands could not be broadcast together with shapes (2,3) (3,1)

2.矩陣乘法

如果真正想要算矩陣*向量的矩陣乘法，要用np.dot

A4=np.array([[1,2,3],[2,3,4]])
b4=np.array([1,2,3])
np.dot(A4,b4)#dot product
#array([14, 20])

列向量也有類似結(jié)果

A4=np.array([[1,2,3],[2,3,4]])
b4=np.array([[1],[2],[3]])
np.dot(A4,b4)#dot product
# array([[14],
#        [20]])

4.矩陣矩陣乘法

1.直接相乘

同樣，也是對(duì)應(yīng)位置相乘

A4=np.array([[1,2,3],[2,3,4]])
B4=np.array([[1,2,3],[4,5,6]])
A4*B4
# array([[ 1,  4,  9],
#        [ 8, 15, 24]])

有廣播機(jī)制

A4=np.array([[1,2,3],[2,3,4]])
B4=np.array([[1,2,3]])
A4*B4
# array([[ 1,  4,  9],
#        [ 2,  6, 12]])

2.np.dot

需要第一個(gè)的列數(shù)和第二個(gè)的行數(shù)相對(duì)應(yīng)

A4=np.array([[1,2,3],[2,3,4]])
B4=np.array([[1,2,3],[4,5,6]])
np.dot(A4,B4.T)
# array([[14, 32],
#        [20, 47]])
 
A5=np.array([[1,2,3],[2,3,4]])
B5=np.array([[1,2,3],[4,5,6],[7,8,9]])
np.dot(A5,B5)
# array([[30, 36, 42],
#        [42, 51, 60]])

對(duì)不上會(huì)報(bào)錯(cuò)

A4=np.array([[1,2,3],[2,3,4]])
B4=np.array([[1,2,3],[4,5,6]])
np.dot(A4,B4)
# shapes (2,3) and (2,3) not aligned: 3 (dim 1) != 2 (dim 0)

5.np.dot 和np.matmul的區(qū)別

Numpy中np.dot與np.matmul的區(qū)別

主要參考以上博客。

1.在二維（矩陣中），二者是一致的

2.在三維（張量中），二者有差別。

以原博客中的例子為例

a = np.array([i for i in range(12)]).reshape([2,2,3])
b = np.array([i for i in range(12)]).reshape([2,3,2])
"""
a
[[[ 0  1  2]
  [ 3  4  5]]
 [[ 6  7  8]
  [ 9 10 11]]]
b
[[[ 0  1]
  [ 2  3]
  [ 4  5]]
 [[ 6  7]
  [ 8  9]
  [10 11]]]
"""

np.dot很清晰，就是a的每一行分別和b的兩層乘起來(lái)，于是2*2輸出了四個(gè)“矩陣”（表示成4維的常數(shù)）：

 
"""
a11= [ 0  1  2]
a12= [ 3  4  5]
a21= [ 6  7  8]
a22= [ 9 10 11]
b
[[[ 0  1]
  [ 2  3]
  [ 4  5]]
 [[ 6  7]
  [ 8  9]
  [10 11]]]
c[:,i,j]=aij*b
"""

如：

[ 10, 13] =[0 1 2]*[[ 0 1]
[ 2 3]
[ 4 5]]
[ 28, 31]=[0 1 2]*[[ 6 7]
[ 8 9]
[ 10 11]]

>>> np.dot(a,b)
array([[[[ 10,  13],
         [ 28,  31]],
 
        [[ 28,  40],
         [100, 112]]],
 
 
       [[[ 46,  67],
         [172, 193]],
 
        [[ 64,  94],
         [244, 274]]]])
>>> np.dot(a,b).shape
(2, 2, 2, 2)

np.matmul的結(jié)果:

>>> np.matmul(a,b)
array([[[ 10,  13],
        [ 28,  40]],
 
       [[172, 193],
        [244, 274]]])
>>> np.matmul(a,b).shape
(2, 2, 2)

可以看出，如果把np.dot視為8行、matmul視為4行的話，matmul正好取第1、3、6、8四行，也就是第一層的前兩行和第二層的后兩行……

直觀理解，ok

到此這篇關(guān)于python中使用numpy包的向量矩陣相乘np.dot和np.matmul實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)numpy向量矩陣相乘np.dot和np.matmul內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: