np.einsum求一个矩阵的立方

可以使用以下代码使用np.einsum来计算矩阵的立方：

import numpy as np

matrix = np.array([[1, 2], [3, 4]])

cube = np.einsum('ij,jk,kl->il', matrix, matrix, matrix)

print(cube)

输出：

[[ 37  54]
 [ 81 118]]

注意，这里使用了`'ij,jk,kl->il'`来指定`np.einsum`的运算规则，该规则表示将第一个矩阵与第二个矩阵的乘积再与第三个矩阵的乘积做乘积，并将结果存储在一个新的矩阵中。