pytorch 多维矩阵相乘（爱因斯坦求和约定）

torch.einsum(equation, *operands) → Tensor函数 该函数提供了一种使用爱因斯坦求和约定来计算多线性表达式（即乘积和）的方法。code Parameters:排序 等式（ string ） - 该等式根据与操做数和结果的每一个维度相关联的小写字母（索引）给出。左侧列出了操做数尺寸，以逗号分隔。每一个张量维度应该有一个索引字母。右侧跟在->以后，并给出输出

>>阅读原文<<