Numpy中einsum函数的一个bug

爱因斯坦求和非常强大.

首先给出一个三阶张量

>>> A = np.array([
... [[111,112],[121,122]],
... [[211,212],[221,222]]
... ])

正常的写法计算出的是正确的结果

>>> np.einsum('ijk->kij',A)
array([[[111, 121],
        [211, 221]],

       [[112, 122],
        [212, 222]]])

简化后的写法会得到错误的答案

>>> np.einsum('kij',A)
array([[[111, 211],
        [112, 212]],

       [[121, 221],
        [122, 222]]])