4 ms·
Some short answers to this and following questions: 1. Einsum doesn’t AFAIK allow custom strides. So Einsum does only the mapreduce part (and that too only for
by ssivark 5y ago
Some short answers to this and following questions:
1. Einsum doesn’t AFAIK allow custom strides. So Einsum does only the mapreduce part (and that too only for multiplication & summation combo). To get a taste for the absolute brilliance of a slightly more generic interface, check out Tullio in Julia.
2. Tensordot etc use BLAS under the hood IIRC and can therefore parallelize automatically. The Einsum backend is good old fashioned single threaded C code, so it’s fast (and inflexible) but single core.