您即将离开知乎，请注意您的账号和财产安全。
https://stackoverflow.com/questions/67570529/derive-the-gradient-through-torch-topk
