numpy矩阵运算--矩阵乘法

1)元素对应相乘,使用 multiply 函数或 * 运算符来实现

a = np.array([2,2,2])
b = np.array([3,3,3])
c1 = a*a 
c1

array([4, 4, 4])
c2 = np.multiply(a,b)
c2

array([6, 6, 6])

2)矩阵相乘,使用 dot函数或 @运算符来实现

a = np.array([[2,2,2],[1,1,1]])    #a.shape(2, 3)
c = np.array([[3,4],[1,0],[1,1]])  #c.shape(3, 2)
b2 = a@c 
b2  

array([[10, 10],[ 5, 5]])          #b2.shape(2, 2)

如果两个一维数组相乘:

a = np.array([2,2,2])
c = np.array([3,4,5])
b1 = np.dot(a,c)       #a@c
b1

24

b2 = a@c
b2

24
 

猜你喜欢

转载自www.cnblogs.com/tongtong123/p/10642817.html