矢量化向量化切片矩阵乘法

Question

我分别具有一个128X3X3和一个128X3X5 3D NumPy矩阵A和B我想将A的每个3by3切片乘以B的每个3by5切片，以最终得到尺寸为128X3X5的矩阵C。当然，所有这些都必须完全矢量化，并且不会产生循环。谢谢！

Answer 1

似乎您想要matrix product（因为您不能仅以放置方式直接将形状不同的矩阵相乘）。因此，您可以使用@运算符：

A@B

或带有np.matmul：

np.matmul

检查：

np.matmul(A, B)