python np.dot()函数的用法
numpy.dot(a,b)
- 如果a和b都是一维数组,那么它返回的就是向量的内积。
import numpy as np
c = np.arange(0,9)
d = c[::-1]
np.dot(c,d)
Out[35]: 84
- 如果a和b都是二维数组,那么它返回的是矩阵乘法。
import numpy as np
a = [[1, 0], [0, 1]]
b = [[4, 1], [2, 2]]
np.dot(a, b)
Out[1]:
array([[4, 1],
[2, 2]])
需要注意的是,np.dot(a,b) 和 np.dot(b,a) 得到的结果是不一定相同的。
参考文档: