any equivalent to np.tensordot? #1517

Activity

1. nilgoyette

   added

   good first issueA good issue to start contributing to ndarray!A good issue to start contributing to ndarray!

   on Jun 3, 2025
2. nilgoyette commented on Jun 3, 2025

   

   nilgoyette

   on Jun 3, 2025

   Collaborator

   More actions

   I don't think we have that function. IIUC, this is not strictly related to ML so I think it would fit in ndarray.
3. hugolgst commented on Jun 14, 2025

   

   hugolgst

   on Jun 14, 2025

   Author

   More actions

   there's https://crates.io/crates/ndarray_einsum_beta
4. jmg049 commented on Nov 8, 2025

   

   jmg049

   on Nov 8, 2025

   Contributor

   More actions

   I'm preparing a PR for this right now. I have the implementation ready and just need to focus on some tests, docs and where exactly to put the functionality.

   I think in my fork I have it under linalg and impl_linalg.rs

   Generally speaking I think there's probably a few more functions like tensordot and also the phase angle functionality I have in another PR.

   I think they're a low hanging fruit of sorts since we have solid reference implementations (Numpy, etc) and they are mostly self contained and do not interfere with broader functionality beyond making the codebase a bit bigger and running some more tests.

   But adding them provides some benefit to users who find themselves reaching for them.
5. jmg049

   added a commit that references this issue on Nov 8, 2025

   Tensordot implementation [Issue rust-ndarray#1517]

   ...

   97ed17b