爱因斯坦求和einsum

numpy中有一个einsum，表示爱因斯坦求和，方便用记号的方式对矩阵或者数组等进行线性代数的操作。

einops
einops包提供了类似的详细接口以涵盖其他操作：转置、重塑/展平、重复/平铺、挤压/取消挤压和缩减。

opt_einsum
opt_einsum以与后端无关的方式优化类似 einsum 的表达式的收缩顺序。

以上3个软件包是常用的3种，在神经网络和线性代数中经常会见到。