PYTORCH torch.einsum 函数

66 阅读 0 评论 44 点赞

我是靠谱客的博主要减肥大地，最近开发中收集的这篇文章主要介绍PYTORCH torch.einsum 函数，觉得挺不错的，现在分享给大家，希望可以做个参考。

以爱因斯坦求和的形式任意定义想要进行的矩阵乘法的操作，可以内部指定输出转置，功能多样，形式灵活。

#theta_phi: nxtxg
#g: nxcxg
output = torch.einsum('ntg, ncg->nct', theta_phi, g)
#output: nxcxt

以上就是要减肥大地为你收集整理的PYTORCH torch.einsum 函数的全部内容，希望文章能够帮你解决PYTORCH torch.einsum 函数所遇到的程序开发问题。

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

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