arrayfire / arrayfire/arrayfire-python
Feature: einsum
- Dominant language
- Python
- Stars
- 422
- Forks
- 63
- PR merge metrics
- No merged PRs in 30d
Description
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]]])
```
Contributor guide
No contributing guide indexed for this repository
Research direction
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.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- numpy, python
- Domain
- hpc
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100