输入矩阵a和b,输出a和b的乘积
时间: 2024-02-17 08:04:16 浏览: 57
输入一个矩阵,输出一个结果
好的,可以使用矩阵乘法来实现。假设矩阵a的大小为m x n,矩阵b的大小为n x p,则输出矩阵c的大小为m x p。
下面是一个使用Python实现矩阵乘法的示例代码:
```python
def matrix_multiply(a, b):
m, n = len(a), len(a[0])
p = len(b[0])
c = [[0] * p for _ in range(m)]
for i in range(m):
for j in range(p):
for k in range(n):
c[i][j] += a[i][k] * b[k][j]
return c
```
其中,a和b分别表示输入的矩阵,c表示输出的矩阵。代码中使用了三重循环来实现矩阵乘法,时间复杂度为O(mnp)。
阅读全文