numpy中有一个einsum,表示爱因斯坦求和,方便用记号的方式对矩阵或者数组等进行线性代数的操作。
einops
einops包提供了类似的详细接口以涵盖其他操作:转置、重塑/展平、重复/平铺、挤压/取消挤压和缩减。
opt_einsum
opt_einsum以与后端无关的方式优化类似 einsum 的表达式的收缩顺序。
以上3个软件包是常用的3种,在神经网络和线性代数中经常会见到。
numpy中有一个einsum,表示爱因斯坦求和,方便用记号的方式对矩阵或者数组等进行线性代数的操作。
einops
einops包提供了类似的详细接口以涵盖其他操作:转置、重塑/展平、重复/平铺、挤压/取消挤压和缩减。
opt_einsum
opt_einsum以与后端无关的方式优化类似 einsum 的表达式的收缩顺序。
以上3个软件包是常用的3种,在神经网络和线性代数中经常会见到。