differentiable top k indices

[Chen-Cai-OSU]

Hello,

I am looking for the the work around of back prop through torch.topk . torch.topk return values and indices. It can backprop values but not indices (see here). I was wondering can torchsort can somehow achieve this? Thank you so much!

[teddykoker]

Hi Chen Cai,

That can be achieved, see #19, specifically this comment: #19 (comment)

Let me know if you need any clarification!