Numpy 沿AIX的两个3D矩阵上的2D点积

给定两个维数分别为（x，y，z）和（y，x，z）的矩阵A和B，如何在两个矩阵的前二维上点积？结果的维数应为（x，x，z）

谢谢

与字面上相同的字符串表达式一起使用-

np.einsum('xyz,yiz->xiz',a,b) # a,b are input arrays

---

请注意，我们使用了

yiz

作为第二个数组的字符串表示法，而不是

yxz

，因为

i

应该是输出数组中的一个新维度，并且不与我们已经为其分配了

x

的第一个数组的第一个轴对齐。要对齐的尺寸标注采用相同的字符串表示法

发布的解决方案对你有用吗？是的！非常感谢你！