pytorch深度学习实战

3.4命名张量

einsum方法详解（爱因斯坦求和）

einsum是pytorch、numpy中一个十分优雅的方法，如果利用得当，可完全代替所有其他的矩阵计算方法，不过这需要一定的学习成本。本文旨在详细解读einsum方法的原理，并给出一些基本示例。
一、爱因斯坦求和

 爱因斯坦求和是一种对求和公式简洁高效的记法，其原则是当变量下标重复出现时，即可省略繁琐的求和符号。

比如求和公式：

 

其中变量 a 与变量 b 的下标是相同的（即重复出现），则可将其记为：

 

二、einsum方法原理

einsum方法正是利用了爱因斯坦求和简介高效的表示方法，从而可以驾驭任何复杂的矩阵计算操作。基本的框架如下：

C