【学习笔记】einsum函数

174 阅读 0 评论 115 点赞

我是靠谱客的博主顺利篮球，这篇文章主要介绍【学习笔记】einsum函数，现在分享给大家，希望可以做个参考。

einsum全称Einstein summation convention（爱因斯坦求和约定），又称为爱因斯坦标记法。einsum的写法省去了求和符号，显得更加简洁。

函数使用

实现 $c_{jk}=sum_ia_{ijk}$

c = a.sum(axis=0)           # 常规
c = np.einsum('ijk->jk', a) # einsum

更高维向量的求和，实现 $c_{*}=sum_ia_{i*}$ ，可以直接将剩下的维度替换成省略号。

c = np.einsum('i...->...', a) # einsum

实现 $c_{ik}=sum_ja_{ij}b_{jk}$

c = np.dot(a, b)                 # 常规
c = np.einsum('ij,jk->ik', a, b) # einsum

更高维张量的乘法，实现 $c_{kl}=sum_isum_ja_{ijk}b_{ijl}$

c = np.einsum('ijk,jkl->kl', a, b) # einsum

以上就是顺利篮球最近收集整理的关于【学习笔记】einsum函数的全部内容，更多相关【学习笔记】einsum函数内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。