NumPy numpy.dot 函式
Suraj Joshi
2023年1月30日
Python Numpy numpy.dot()
函式計算兩個輸入陣列的點積。
numpy.dot()
語法
numpy.dot(a, b, out=None)
引數
a |
類陣列。計算點積的第一個陣列或標量 |
b |
類陣列。計算點積的第二個陣列或標量 |
out |
陣列。一個可選的引數,其資料型別必須與輸出的預期資料型別相同。 |
返回值
它返回輸入向量的點積。如果兩個輸入都是標量,則產生一個 1 維陣列,否則產生 n 維陣列。
如果第 1 個輸入陣列的最後一個維度不等於第 2 個輸入陣列的第 2 到最後一個維度,會引發 ValueError
。
示例程式碼:numpy.dot()
查詢點積的方法
當兩個輸入均為一維陣列時
import numpy as np
a=4
b=5
prod=np.dot(a,b)
print(prod)
輸出:
20
這裡,由於 a
和 b
都是一維陣列,np.dot()
函式只是返回一個標量,也就是兩個數字的簡單乘積。
當兩個輸入都是向量時
import numpy as np
a=np.array([3,4])
b=np.array([4,5])
prod=np.dot(a,b)
print(prod)
輸出:
32
它計算向量的點積。
兩個向量 [x1,y1]
和 [x2,y2]
的點積由 x1*x2+y1*y2
給出。
當兩個輸入均為二維陣列時
import numpy as np
a=np.array([[3,4],
[2,3]])
b=np.array([[4,5],
[2,3]])
prod=np.dot(a,b)
print(prod)
輸出:
[[20 27]
[14 19]]
它計算矩陣的乘積。
作者: Suraj Joshi
Suraj Joshi is a backend software engineer at Matrice.ai.
LinkedIn