NumPy numpy.dot 函式

Suraj Joshi 2023年1月30日 2020年11月7日
  1. numpy.dot() 語法
  2. 示例程式碼:numpy.dot() 查詢點積的方法
NumPy numpy.dot 函式

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

這裡,由於 ab 都是一維陣列,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]]

它計算矩陣的乘積。

Author: Suraj Joshi
Suraj Joshi avatar Suraj Joshi avatar

Suraj Joshi is a backend software engineer at Matrice.ai.

LinkedIn