pytorch 多維矩陣相乘（愛因斯坦求和約定）

torch.einsum(equation, *operands) → Tensor函數 該函數提供了一種使用愛因斯坦求和約定來計算多線性表達式（即乘積和）的方法。code Parameters:排序 等式（ string ） - 該等式根據與操做數和結果的每一個維度相關聯的小寫字母（索引）給出。左側列出了操做數尺寸，以逗號分隔。每一個張量維度應該有一個索引字母。右側跟在->以後，並給出輸出

>>阅读原文<<