4 ms·
The author posted a previous article about why they don't like numpy and his problems with einsum: https://dynomight.net/numpy/ https://dynomight.net/numpy/
by sottol 1y ago
The author posted a previous article about why they don't like numpy and his problems with einsum:
https://dynomight.net/numpy/ https://dynomight.net/numpy/
- adgjlsfhk1 1y agoI think you could for linear solves into a "generalized Einstein notation", but the other option you have is to support more complex array types, in which case a batched linearsolve can be reframed as a single linearsolve by a block-diagonal matrix