That is true. I’m trying to implement the result of a paper that describes a method to reasonably approximate a backward pass, so even if you have a discrete function like argsort I can define an alternative derivative through which it makes sense to backpropagate. I know the example doesn’t make a lot of sense like this, though. However, I figured there has to be a way to get this working. I don’t really care if this is the proper derivative of argsort.