一. 爱因斯坦求和torch.einsum()链接 einsum is all you neededeinsum初探 : “输入标记中重复字母表示沿这些轴的值将相乘”,“输出标记中省略字母表示沿该轴的值将被求和”Pytorch中, torch.einsum详解。一文学会 Pytorch 中的 einsum 二. torch.einsum()重要规则