[Task] Generalize TopK to all model types doing multiclass classification in Merlin Models #1050
Description
add a topK operation on top of softmax layer.
Metadata
Metadata
Assignees
Labels
No labels
{{ message }}
add a topK operation on top of softmax layer.