【Pytorch】einsum

本文为 Pytorch 学习笔记，讲解爱因斯坦简记法。欢迎在评论区与我讨论👏

爱因斯坦简记法

是爱因斯坦提出的对向量、矩阵、张量求和运算的简记法。

被省略的部分是求和符号和求和号的下标。

省略规则：默认成对出现的下标为求和下标。

举例

1. 用 $x_i,y_i$ 简化表示内积 $<x,y>$：
   $$x_i{y}_i:=\sum _i{x}_i{y}_i=o$$
   其中 o