I found myself in a loud coffee shop unable to work on other stuff so I picked off #72. I am not sure if the custom combinator code is needed elsewhere but itertools combinations with replacement should be an appropriate replacement with the advantage that it doesn't have to actually generate all the index sets in memory. On my laptop I didn't see any major improvement for this because N>=8, M>=8 had a huge result tensor anyway.
I found myself in a loud coffee shop unable to work on other stuff so I picked off #72. I am not sure if the custom combinator code is needed elsewhere but itertools combinations with replacement should be an appropriate replacement with the advantage that it doesn't have to actually generate all the index sets in memory. On my laptop I didn't see any major improvement for this because N>=8, M>=8 had a huge result tensor anyway.
:books: Documentation preview :books:: https://pyttb--222.org.readthedocs.build/en/222/