如果问pytorch中最强大的一个数学函数是什么？
我会说是torch.einsum：爱因斯坦求和函数。
它几乎是一个"万能函数"：能实现超过一万种功能的函数。
不仅如此，和其它pytorch中的函数一样，torch.einsum是支持求导和反向传播的，并且计算效率非常高。
einsum 提供了一套既简洁又优雅的规则，可实现包括但不限于：内积，外积，矩阵乘法，转置和张量收缩（tensor contraction）等张量操作，熟练掌握 einsum 可以很方便的实现复杂的张量操作，而且不容易出错。
尤其是在一些包括batch维度的高阶张量的相关计算中，若使用普通的矩阵乘法、求和、转置等算子来实现很容易出现维度匹配等问题，但换成einsum则会特别简单。
套用一句深度学习paper标题当中非常时髦的话术，torch.einsum is all you needed
