沿特定尺寸的矩阵积

Question

考虑d维数组B（II1，…，IYD）的大小（n1…x）…x N_D）和大小为（N_D x N_D）的二维数组O^D。 我需要计算一下

A（i_1，…，i_d，…），i_D）=和{j_D=1}^{N_D}O_{i_D，j_D}B（i_1，…，j_D，…，i_D）

以最有效的方式，因为A和B阵列将很大。 换句话说，这基本上是一个矩阵乘法积O.B，其中一个沿着O的第二维和B的第d维收缩，但重要的是，最终结果中的索引也按（i_1…i_d）而不是（i_d i_1。。。我不知道。 我通过一系列的换位来实现这一点，但它似乎并不特别整洁或有效；我想知道是否有更优雅/自动的方法来做同样的事情