I think this functionality is available in OMEinsum and (possibly without the derivs) in TensorOperations. Both packages need specifying the dimensions explicitly, though.
tanhevg
4
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Is there a function compute matrix multiplication in a multimensional matrix? | 8 | 1511 | December 3, 2018 | |
| Matrix multipcation broadcasting? | 3 | 387 | June 17, 2023 | |
| Matrix multiplication on multiple dimensions | 8 | 4223 | October 28, 2020 | |
| Batched matmul with einsum | 1 | 169 | June 11, 2024 | |
| Multiply many-matrices by many-vectors | 33 | 6944 | December 14, 2018 |