爱因斯坦简记法:是一种由爱因斯坦提出的,对向量、矩阵、张量的求和运算简单记录方法
使用和简化表示矩阵乘法XY
import numpy as np
a=np.arange(60).reshape(3,4,5)
b=np.arange(24).reshape(4,3,2)
o = np.einsum('ijk,jil->kl', a, b)
array([[4400, 4730],
[4532, 4874],
[4664, 5018],
[4796, 5162],
[4928, 5306]])