如何使用 numpy 轻松执行随机矩阵乘法？

Question

我想生成 2 个随机 3x4 矩阵，其中条目呈正态分布，A 和 B。之后，我有一个 2x2 矩阵 C = [[a,b][c,d]]，我想使用它生成 2 个新的 3x4 矩阵 A' 和 B'，其中 A' = a A + b B，B' = c A + d B。

为了生成矩阵 A 和 B，我正在考虑使用这行代码：

Z = np.random.normal(0.0, 1.0, [2,3, 4])

但是，给定矩阵 C，我不知道如何使用简单的 Numpy 向量化来实现矩阵 A' 和 B'，或者等效地，实现包含 A' 和 B' 的 2x3x4 数组。有什么想法吗？

Answer 1

我想你可以使用

np.einsum

np.einsum("ij, jkl -> ikl", C, Z)