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)会完全满足您的要求.
标签:linear-algebra,numeric,python,numpy 来源: https://codeday.me/bug/20191030/1971423.html