如何计算点积矩阵？

Question

所以，我有两个三维数组，都有shape (n, m, n)。我想计算我的数组所包含的向量的点积的2d矩阵（(n, m)）。像这样的东西，但最好是矢量化的：

res = np.zeros(a.shape[:-1])

for i in range(len(a)):
    for j in range(len(a[0])):
        res[i][j] = a[i][j] @ b[i][j]

我已经用dot和tensordot摆弄了大约一个小时，但是根本无法解决这个问题！