python矩阵乘法函数
时间: 2023-11-07 19:48:51 浏览: 107
在Python中,矩阵乘法可以使用NumPy库的np.dot()函数进行计算。np.dot()函数接受两个NumPy数组作为参数,并返回数组的乘积。对于两个矩阵A和B,A中每行的元素值与B中每列的元素值对应相乘再求和,得到结果矩阵C的一个元素值。其中A的维度为(m,n),B的维度为(n,c),则A和B相乘后得到的矩阵C的维度为(m,c)。
使用np.dot()函数可以进行向量点积和矩阵乘法。格式为x.dot(y)或np.dot(x, y)。如果x是m×n矩阵,y是n×m矩阵,则x.dot(y)得到m×m矩阵。
以下是一个Python示例,展示了向量相乘和矩阵相乘的使用方法:
向量相乘,得到内积:
```python
import numpy as np
x = np.array([0, 1, 2, 3, 4])
y = x[::-1]
result = np.dot(x, y)
print(result) # 输出结果: 10
```
矩阵相乘,得到矩阵的积:
```python
import numpy as np
x = np.arange(0, 6).reshape(2, 3)
y = np.random.randint(0, 10, size=(3, 2))
result = np.dot(x, y)
print(result) # 输出结果: [[12 19], [42 73]]
```
需要注意的是:A的列和B的行的维度必须保持一致,否则会报错。同时,np.dot(A, B)和np.dot(B, A)的值可能不一样,操作数(A、B)的顺序不同,结果也会不同。
阅读全文