NumPy numpy.dot 函数
Suraj Joshi
2023年1月30日
NumPy
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]]
它计算矩阵的乘积。
Enjoying our tutorials? Subscribe to DelftStack on YouTube to support us in creating more high-quality video guides. Subscribe
作者: Suraj Joshi
Suraj Joshi is a backend software engineer at Matrice.ai.
LinkedIn