python - python中多个张量的一般乘积

Question

在python中（最好在numpy数组框架下），一次做多个张量乘积的最好方法是什么，而不是使用numpy.tensordot一个一个地做？假设我们需要做这个求和：

sum over (i,j,k) of :  A[i,j] * B[j,k,l] * C[k,i]

其中总和超过 i,j,k，并且 l 故意不求和 - 因此我们希望最终有一个一维向量。

Answer 1

退房einsum。我认为np.einsum('ij,jkl,ki', A, B, C)会做你想要的。