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