orion
orion copied to clipboard
Einsum
Feature Request
Describe the Feature Request
Tensor contraction over specified indices and outer product.
Here is a full description of Einsum in ONNX: https://github.com/onnx/onnx/blob/main/docs/Operators.md#einsum
Additional Context
You can take inspiration to EZKL or Julia-lang implementation of Einsum:
- https://github.com/zkonduit/ezkl/blob/bc5abf50a23bc00b95e133d380c5b0926a1528fd/src/tensor/ops.rs#L367
- https://github.com/ahwillia/Einsum.jl