如何在numpy中进行矩阵乘法？

Question

我有两个numpy ndarray，形状为X的{​​{1}}和形状为(b, m)的{​​{1}}。

我想创建一个形状为Y的新的numpy ndarray (b, m, n)，其中：

Z

例如，Z的第一行（索引0）应该是X和Y [0]的乘法的第一行（索引0）。

这里是执行此操作的一种方法：

(b, n)

此方法的问题是存在浪费的未使用的计算。另外，我们必须顺序堆叠最终矩阵Z[i] = np.matmul(X, Y[i])[i]。

有更好的方法吗？