numpy矩阵乘法
时间: 2023-07-09 20:25:06 浏览: 104
使用Numpy/TensorFlow中的tensordot进行多维矩阵相乘
在NumPy中,可以使用`numpy.dot()`函数进行矩阵乘法。示例如下:
```python
import numpy as np
# 创建两个矩阵
A = np.array([[1, 2], [3, 4]])
B = np.array([[5, 6], [7, 8]])
# 矩阵乘法
C = np.dot(A, B)
print(C)
```
输出结果为:
```
[[19 22]
[43 50]]
```
其中,矩阵A和矩阵B的乘积为矩阵C。如果需要对于多个矩阵进行乘法运算,可以使用`numpy.matmul()`函数。
阅读全文