RFC: add support for batched matmul / einsum / batched tensordot
还没有人认领这个 Issue。
评估
- 难度
- 5/5
- 预计耗时
- 一周以上
- 新手友好度
- 25/100
- Issue 类型
- 功能
- 描述清晰度
- 需要澄清
- 活跃度
- 停滞
- 领域
- data
调研方向
该 issue 未指定任何文件、测试或入口点。首先检查 Array API 标准，并比较提议的 batched_matmul、batched tensordot 和 einsum 方法；完成的标准是选择并明确规定一个受支持的操作，而不是保留多个未决的替代方案。
由索引模型根据 Issue 内容生成。
描述
Currently I don't see a way to implement einsum or batched matmul using the standard. There is a way with explicit loop, but that does not count.
Some ways this can be addressed:
- minimal implementation would include batched_matmul (signature
bij, bjk -> bik). Other operations can be reduced to it with reshapes. This may include additional copies - modification of tensordot to include batched dimensions thus should be more efficient, but AFAIK this method is not surfaced in any major framework
- include einsum in the standard
- 主要语言
- Python
- 星标
- 281
- 派生
- 52
- PR 合并指标
- 30 天内没有已合并 PR
贡献指南
从这里开始
- 先读完整个 Issue，再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库，在一个分支上完成修改。
- 提交 Pull Request，并在描述里引用这个 Issue 编号。
data-apis/array-api 的其他 Issue
-
难度 1/5 1-3 小时 新手友好度 88/100
-
难度 2/5 1-3 小时 新手友好度 70/100
-
难度 5/5 一周以上 新手友好度 35/100
-
Maintenance
-
难度 5/5 一周以上 新手友好度 25/100
查看 data-apis/array-api 的全部 Issue
相似的 Issue
-
难度 2/5 1-3 小时 新手友好度 88/100
-
难度 2/5 1-3 小时 新手友好度 82/100
-
难度 2/5 1-3 小时 新手友好度 78/100
-
enhancement
难度 2/5 1-3 小时 新手友好度 72/100
-
难度 2/5 1-3 小时 新手友好度 74/100