会员
周边
捐助
新闻
博问
闪存
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
嶙羽
博客园
首页
新随笔
联系
订阅
管理
2021年3月4日
einsum爱因斯坦求和
摘要: 最近因为vision transformer里的pytorch代码,看到了torch.einsum(np.einsum同理)这个操作,简直是神了; 比如 t = torch.randn(2,4,3) q, k, v = tuple(rearrange(t, 'b t (d k) -> k b t d
阅读全文
posted @ 2021-03-04 13:31 嶙羽
阅读(332)
评论(0)
推荐(0)
编辑
公告