Skip to content
New issue

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

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

Already on GitHub? # 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
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
feature The feature request
Projects
None yet
Development

No branches or pull requests

2 participants