冷傲咖啡

文章
7
资源
0
加入时间
2年10月17天

pytorch einsum, numpy einsum什么是einsum?为什么用?具体例子

什么是einsum?爱因斯坦求和约定:https://zhuanlan.zhihu.com/p/101157166https://en.wikipedia.org/wiki/Einstein_notation为什么用?简洁, 强大具体例子pytorch和numpy一样的,这里以pytorch为例.矩阵乘矩阵, C = A x Beg1. (ij, jk -> ik), 实际上就是[C]ik=∑j[A]ij×[B]jk[C]_{ik} = \sum_{j} [A]_{ij} \t