Thank you, that looks promising!
Any idea how long it would take before it makes it into torch? (weeks?/months?/longer?)
I tried to just implement the changes in that file, but unfortunately it seems that there are quite a few others, branching to other files as well.
For now I will put it on the shelf, and just work with builtin functions for now (which will be about 10 times slower, but since my code isn’t ready for large scale tests anyway that should be fine for now).
But I’m really hoping this will fix it down the road.