python通过numpy实现两个矩阵的乘法
在 numpy 中，使用 numpy.dot() 函数进行矩阵乘法。下面是一个示例代码：
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)
在上面的代码中，我们首先定义了两个 2x2 的矩阵 A 和 B，然后使用 numpy.dot() 函数计算它们的乘积，结果存储到 C 中。最后，我们将 C 打印输出。
如果你使用的是 numpy 1.13 以及以上版本，还可以使用 @ 符号进行矩阵乘法：
import numpy as np # 定义两个矩阵 A = np.array([[1, 2], [3, 4]]) B = np.array([[5, 6], [7, 8]]) # 计算矩阵乘法 C = A @ B print(C)
上面的代码和之前的示例代码等价。
相关文章