Python中多个张量的一般乘积

python中(最好是在numpy数组框架下),最好是一次处理多个张量的乘积而不是使用numpy.tensordot一张一张地做的最佳方法?假设我们需要进行以下总结:

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

其中总和超过i,j,k和l的总和不是故意求和的-因此我们期望最后有一个一维矢量.

解决方法:

签出einsum.
我认为np.einsum(‘ij,jkl,ki’,A,B,C)会完全满足您的要求.

上一篇:攻防世界simple_php


下一篇:Javascript数字比较运算符如何精确地处理字符串?