详细内容或原文请订阅后点击阅览
理解爱因斯坦符号和 einsum 乘法
使用字符串符号执行高阶张量运算继续阅读 Towards Data Science »
来源:走向数据科学了解爱因斯坦的符号和Einsum乘法
用字符串符号执行高阶张量操作
简介
机器学习在很大程度上取决于线性代数操作,至关重要的是要很好地了解矩阵(或更精确地之间的张量)之间的乘法如何工作。
我经常注意到那些接近pytorch或Tensorflow等机器学习框架的人在遇到有关张量之间尺寸不匹配的错误时存在许多问题。在这篇简短的文章中,我想对此提出一些清晰度,并介绍使用Einsum。
einsum从标量到张量
在机器学习中,我们根据数据的维度(DIM)区分数据。使用:
- 标量dim = 0向量= 1matrices dim = 2tensor dim = 3(或更多)
在Pytorch中,我们可以很容易地创建这些数据类型。
导入火炬#创建标量张量表…