aesara
aesara copied to clipboard
Implement `einsum` equivalent
Probably quite some work, but would be nice to have an equivalent: https://numpy.org/doc/stable/reference/generated/numpy.einsum.html
It seems like there is a C-API function for this as well: https://numpy.org/devdocs/reference/c-api/array.html#c.PyArray_EinsteinSum
This might be relevant: https://optimized-einsum.readthedocs.io/en/stable/backends.html