Davi Silva 7fbebb3df6 Implement einsum (#2686) ...

* hopeful impl for Tensor.einsum

* satisfy mypy by having less typing. :(

* a few simple tests

* even more tests

* permute tests

* xfails for improper usage

* fix LLVM test fail

* use argfix

* more helpful error message on shape mismatch

2023-12-10 15:56:01 -08:00

78 KiB

Raw Blame History

View Raw