numpy多个矩阵乘法
时间: 2023-10-08 18:10:26 浏览: 78
多个矩阵相乘
5星 · 资源好评率100%
在 NumPy 中,可以使用 `dot()` 方法来计算两个矩阵的乘积。对于多个矩阵的乘积,可以使用 `reduce()` 函数和 `dot()` 方法结合起来实现。
下面是一个例子,其中有三个矩阵 A、B 和 C,需要计算它们的乘积:
```python
import numpy as np
A = np.array([[1, 2],
[3, 4]])
B = np.array([[5, 6],
[7, 8]])
C = np.array([[9, 10],
[11, 12]])
# 计算 A * B * C
result = np.dot(A, np.dot(B, C))
print(result)
```
输出结果为:
```
[[337 386]
[733 842]]
```
注意,使用 `reduce()` 函数时,需要指定 `dot()` 方法的初始值为 `1`,否则会报错。完整的代码如下:
```python
import numpy as np
A = np.array([[1, 2],
[3, 4]])
B = np.array([[5, 6],
[7, 8]])
C = np.array([[9, 10],
[11, 12]])
# 计算 A * B * C
result = np.multiply.reduce([A, B, C], axis=0)
print(result)
```
阅读全文