NumPy numpy.dot 函数

Suraj Joshi 2023年1月30日
  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]]

它计算矩阵的乘积。

作者: Suraj Joshi
Suraj Joshi avatar Suraj Joshi avatar

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

LinkedIn