arrayfire / arrayfire/arrayfire-python
Feature: einsum
Aberta
- Linguagem predominante
- Python
- Estrelas
- 422
- Forks
- 63
- Métricas de merge de PRs
- Nenhum PR com merge em 30d
Descrição
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]]])
```
Guia de contribuição
Nenhum guia de contribuição indexado para este repositório
Avaliação
Esta issue ainda não foi avaliada.