arrayfire / arrayfire/arrayfire-python
Feature: einsum
Aperta
- Lingua principale
- Python
- Stelle
- 422
- Fork
- 63
- Metriche di merge delle PR
- Nessuna PR unita negli ultimi 30g
Descrizione
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]]])
```
Guida per i contributori
Nessuna guida per i contributori indicizzata per questo repository
Valutazione
Questa issue non è ancora stata valutata.