arrayfire / arrayfire/arrayfire-python
Feature: einsum
- 主要语言
- Python
- 星标
- 422
- 派生
- 63
- PR 合并指标
- 30 天内没有已合并 PR
描述
Good day!
Could you please Implement something like numpy.einsum (for OpenCL and CUDA)?
In particular, I'm interested in row/column-wise outer product, like this, but einsum is very universal and useful thing...
```
A = np.random.randint(1,7, size=(3,2))
B = np.random.randint(1,7, size=(2,3))
X = np.einsum('ik,kj->kij', A,B)
X
array([[[ 8, 16, 12],
[ 8, 16, 12],
[ 4, 8, 6]],
[[15, 3, 12],
[ 5, 1, 4],
[ 5, 1, 4]]])
```
贡献指南
这个仓库没有索引到贡献指南
调研方向
The issue names no repository files, tests, or entry points; begin by locating the Python binding's array-operation API and reviewing how OpenCL and CUDA operations are exposed. Compare the requested NumPy einsum behavior and example with existing capabilities, then define supported cases and tests before implementation.
由索引模型根据 Issue 内容生成。
评估
- 技术栈
- numpy, python
- 领域
- hpc
- Issue 类型
- 功能
- 难度
- 5/5
- 预计耗时
- 一周以上
- 活跃度
- 停滞
- 描述清晰度
- 需要澄清
- 新手友好度
- 20/100