如何在numpy中向量化此循环？

Question

我正在尝试将此表达式矢量化：

np.vstack([np.dot(arr3d[k], arr2d.T[k]) for k in range(arr3d.shape[0])]). 

这是matrix.vector到cube。（向量矩阵）的扩展

我可以用一个numpy索引替换k循环以实现相同的操作吗？ arr3d是一个NxNxN数组，而arr2d是一个NxN数组。

我尝试使用np.einsum和np.indices失败

我确切地期望np.vstack([np.dot(arr3d[k], arr2d.T[k]) for k in range(arr3d.shape[0])])的结果，即NxN矩阵= NxNxN cube。 （向量的NxN矩阵）