The output topK's K is fixed now. Do you think training a classifier to predict the value of K for every input is a good solution? Thank you very much.