Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement something like torch.einsum for Tensors #2480

Open
TheLostLambda opened this issue Nov 11, 2024 · 0 comments
Open

Implement something like torch.einsum for Tensors #2480

TheLostLambda opened this issue Nov 11, 2024 · 0 comments
Labels
feature The feature request

Comments

@TheLostLambda
Copy link

Feature description

Something like https://pytorch.org/docs/stable/generated/torch.einsum.html, with something similar done in Rust already here: https://docs.rs/einsum-codegen/latest/einsum_codegen/

Feature motivation

It's a more compact way to write out a number of tensor operations that come up throughout ML work.

(Optional) Suggest a Solution

It seems like a good place for writing a macro! It looks like there are some details on the algorithm here: https://docs.rs/einsum-codegen/latest/einsum_codegen/#einsum-algorithm

@laggui laggui added the feature The feature request label Nov 12, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature The feature request
Projects
None yet
Development

No branches or pull requests

2 participants