【学习笔记】einsum函数

84 阅读 0 评论 56 点赞

我是靠谱客的博主顺利篮球，最近开发中收集的这篇文章主要介绍【学习笔记】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函数所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

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