numpy.dot

原型: numpy.dot(a, b, out=None) 兩個numpy陣列的點乘 (1). 如果兩個引數$a,b$都是$1$維的，該運算向量內積（沒有複數共軛時） np.dot([1,2],[1,2]) 5 (2). 如果兩個引數$a,b$都是$2$維的，該運算做矩陣乘法，但是使用matmul或者[email protected]更好 np.dot([[1,2],[3,4]],[[1,0],[0,1]])

array([[1, 2],
       [3, 4]])

(3). 如果引數$a$或者$b$是$0$維（標量）的，該運算與multiply，但是使用np.multiply(a, b)

array([[1, 2],
       [3, 4]])

(4). 如果$a$是$N$維的，$b$是$1$維（標量）的，它做的是$a$在最後一根軸方向上與$b$做點積(相乘求和)。

a = np.arange(3*2).reshape((3,2))
print(a)
print(np.dot(a, [1,1]))

[[0 1]
 [2 3]
 [4 5]]

[1 5 9]

(5). 如果$a$是$N$維的，$b$是$M(M>=2)$維的，它做的是$a$在最後一根軸方向上與$b$