We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
torch.einsum
Tensor
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/
It's a more compact way to write out a number of tensor operations that come up throughout ML work.
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
The text was updated successfully, but these errors were encountered:
No branches or pull requests
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
The text was updated successfully, but these errors were encountered: