我要评价
E = np.einsum('ij,jk,kl->ijkl',A,A,A)
F = np.einsum('ijki->ijk',E)
其中 A 是一个矩阵(大小不超过 1000 x 1000)。计算 E 很慢。我想通过仅计算我存储在 F 中的“对角线”元素来加快速度。是否可以将这两个表达式组合起来？/是否有更好的方法来加速此计算？
最佳答案
我不确定是否有自动方法，但你总是可以自己做数学运算并给 einsum 最终表达式:
F = np.einsum('ij,jk,ki->ijk', A, A, A)
关于python - 组合 Einsum 表达式，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/36879311/